gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1174

Browse Source 
SIENTIAPDE-1174 Update MLFlowRepository initialization to include logger for enhanced logging capabilities
This commit is contained in:
vitor-aignosi
2025-08-01 17:15:49 -03:00
parent 669494c90b
commit 47df5ebc60
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
laborious/activities/mlflow.py
View File

@@ -23,7 +23,7 @@ class MLFlow(BaseActivity):
self.mlflow_password = mlflow_password self.mlflow_password = mlflow_password
self.model_monitoring_repository = MLFlowRepository( 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") @activity.defn(name="request_transform")

5
laborious/utils/repository/model_repository.py
View File

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