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

SIENTIAPDE-1222

Browse Source 
SIENTIAPDE-1222 Update MLFlow methods to include metadata parameter

- Modified the transform and predict method calls in the MLFlow class to include a new 'metadata' parameter, enhancing the functionality and data handling capabilities of the model monitoring repository.
This commit is contained in:
vitor-aignosi
2025-09-17 10:11:54 -03:00
parent b2ef523c3b
commit 863bbdcc57
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
laborious/activities/mlflow.py
View File

@@ -120,7 +120,7 @@ class MLFlow(BaseActivity):
# Request transformation from MLFlow model
response_data = self.model_monitoring_repository.transform(
model_name, data, model_config
model_name, data, model_config, metadata
)
self.debug(
@@ -176,7 +176,7 @@ class MLFlow(BaseActivity):
# Request prediction from MLFlow model
response_data = self.model_monitoring_repository.predict(
model_name, data, model_config
model_name, data, model_config, metadata
)
self.debug(