SIENTIAPDE-1174

SIENTIAPDE-1174 Update MLFlowRepository initialization to include logger for enhanced logging capabilities

laborious/activities/mlflow.py

@@ -23,7 +23,7 @@ class MLFlow(BaseActivity):
         self.mlflow_password = mlflow_password
 
         self.model_monitoring_repository = MLFlowRepository(
-            f"{mlflow_host}:{mlflow_port}", mlflow_username, mlflow_password
+            f"{mlflow_host}:{mlflow_port}", mlflow_username, mlflow_password, logger.base_logger
         )
 
     @activity.defn(name="request_transform")

laborious/utils/repository/model_repository.py

@@ -19,10 +19,11 @@ from sientia.ModelServing import ModelServing
 
 
 class MLFlowRepository():
-    def __init__(self, host, username, password):
+    def __init__(self, host, username, password, logger):
 
         self.model_serving = ModelServing(tracking_uri=host,
-                                          username=username, password=password)
+                                          username=username, password=password,
+                                          logger=logger)
 
     def transform(self, model_name: str, data: pd.DataFrame, model_retention: int):
         """