From 863bbdcc5713f68fb566f6f0b558b665cb5fa2cf Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 17 Sep 2025 10:11:54 -0300 Subject: [PATCH] SIENTIAPDE-1222 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. --- laborious/activities/mlflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 896f05a..28086d5 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -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(