Update train.py

added some logs
Aman123lug · Aug 28, 2023 · c820065 · c820065
1 parent d5dd3d5
commit c820065
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/train.py b/src/train.py
@@ -46,6 +46,7 @@ def training(self, path:Path):
 
             signature = infer_signature(X_train, y_pred)
             tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
+            logger.info(f" MLflow log_metric saved !")
 
             # Model registry does not work with file store
             if tracking_url_type_store != "file":