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(