From 47df5ebc60e9a234fe052e6f754e44e13e05c95e Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Fri, 1 Aug 2025 17:15:49 -0300 Subject: [PATCH] SIENTIAPDE-1174 SIENTIAPDE-1174 Update MLFlowRepository initialization to include logger for enhanced logging capabilities --- laborious/activities/mlflow.py | 2 +- laborious/utils/repository/model_repository.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 4e5e3a6..af53a22 100644 --- a/laborious/activities/mlflow.py +++ b/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") diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index c3fe550..d518750 100644 --- a/laborious/utils/repository/model_repository.py +++ b/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): """