SIENTIAPDE-1241: refactor train_model workflow due to I/O errors.

model_manager/activities/activities.py

@@ -7,12 +7,12 @@ with workflow.unsafe.imports_passed_through():
     from sientia_do.observability.logger import Logger
 
     from model_manager.activities.experiment_tracking import ExperimentTracking
-    from model_manager.activities.minio import MinIO
-    from model_manager.activities.mlflow import MLFlow
     from model_manager.activities.training import Training
+    from model_manager.utils.repository.model_repository import ModelRepository
+    from model_manager.utils.repository.storage_repository import StorageRepository
 
 
-class Activities(ExperimentTracking, MLFlow, MinIO, Training):
+class Activities(ExperimentTracking, Training):
     """
     Main activities orchestrator for the Model Manager system.
 
@@ -75,18 +75,14 @@ class Activities(ExperimentTracking, MLFlow, MinIO, Training):
             notification_handler=notification_handler,
         )
 
-        MLFlow.__init__(
-            self,
-            mlflow_host=mlflow_config['host'],
-            mlflow_port=mlflow_config['port'],
-            mlflow_username=mlflow_config['username'],
-            mlflow_password=mlflow_config['password'],
+        self.model_repository = ModelRepository(
+            url=mlflow_config['url'],
+            username=mlflow_config['username'],
+            password=mlflow_config['password'],
             logger=logger,
-            notification_handler=notification_handler,
         )
 
-        MinIO.__init__(
-            self,
+        self.storage_repository = StorageRepository(
             endpoint_url=minio_config['endpoint_url'],
             access_key=minio_config['access_key'],
             secret_key=minio_config['secret_key'],
@@ -97,10 +93,15 @@ class Activities(ExperimentTracking, MLFlow, MinIO, Training):
             connect_timeout=minio_config['connect_timeout'],
             read_timeout=minio_config['read_timeout'],
             logger=logger,
-            notification_handler=notification_handler,
         )
 
-        Training.__init__(self, logger=logger, notification_handler=notification_handler)
+        Training.__init__(
+            self,
+            model_repository=self.model_repository,
+            storage_repository=self.storage_repository,
+            logger=logger,
+            notification_handler=notification_handler,
+        )
 
     def __del__(self):
         """