From a817d9855be149d6057880aabb97e02ea80455bb Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Mon, 15 Sep 2025 15:16:27 -0300 Subject: [PATCH] SIENTIAPDE-1222 Refactor MLFlow logging to improve data output clarity - Updated the debug logging to directly capture the output of data.to_csv, enhancing traceability of processed input data. - Removed redundant debug statements for raw response data to streamline logging and focus on essential information. --- laborious/activities/mlflow.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index d0d47de..a3dd54a 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -154,18 +154,15 @@ class MLFlow(BaseActivity): data.columns.name = None self.debug("Processed input data:", metadata) - data.to_csv('data.csv') - self.debug(data.to_string(), metadata) + self.debug(data.to_csv('data.csv'), metadata) # Request transformation from MLFlow model response_data = self.model_monitoring_repository.transform( model_name, data, model_config ) - self.debug("Raw response data:", metadata) - self.debug(response_data, metadata) - if response_data['success']: + response_dataframe = DataFrame(response_data['content']) try: response_dataframe = self.detect_and_parse_datetime_index(