app

app.py

@@ -114,6 +114,8 @@ from xgboost import XGBRegressor, DMatrix, train as xgb_train
 from lifelines import CoxPHFitter
 from sksurv.util import Surv
 from sksurv.metrics import concordance_index_censored
+from mlflow import sklearn as mlflow_sklearn
+import mlflow.sklearn
 
 warnings.filterwarnings("ignore", category=UndefinedMetricWarning)
 set_config(display="text")
@@ -131,13 +133,12 @@ def load_raw_data() -> pd.DataFrame:
 # 2) FONCTION D’ENTRAÎNEMENT + PRÉDICTIONS
 # ────────────────────────────────────────────────
 
+
 mlflow.set_tracking_uri(os.getenv("BACKEND_STORE_URI"))  # URI NeonDB
 os.environ["MLFLOW_DEFAULT_ARTIFACT_ROOT"] = os.getenv("MLFLOW_DEFAULT_ARTIFACT_ROOT")  # S3
 os.environ["AWS_ACCESS_KEY_ID"] = os.getenv("AWS_ACCESS_KEY_ID")
 os.environ["AWS_SECRET_ACCESS_KEY"] = os.getenv("AWS_SECRET_ACCESS_KEY")
 
-from mlflow import sklearn as mlflow_sklearn
-import mlflow.sklearn
 
 # ────────────────────────────────────────────────
 # 2) FONCTION DE PREDICTIONS (sans entraînement)