modif pour version sickit-learn

app.py

@@ -7,3 +7,12 @@ app = FastAPI()
 def retrain():
     result = run_retraining()
     return result
+
+@app.get("/version")
+def version():
+    import sklearn
+    import sys
+    return {
+        "python": sys.version,
+        "sklearn": sklearn.__version__
+    }

retrain.py

@@ -53,7 +53,7 @@ def run_retraining():
         # Sauvegarde modèle dans S3
         s3 = boto3.client("s3")
         bucket = os.getenv("S3_BUCKET")
-        key = f"mlflow/models/model_{run.info.run_id}.joblib"
+        key = f"mlflow/models/xgboost_model_{run.info.run_id}.joblib"
 
         buffer = io.BytesIO()
         joblib.dump(xgb_final, buffer)