From b1951ea2c09214f2b250c31b237a08b87c49b04c Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Tue, 16 Sep 2025 11:54:35 -0300 Subject: [PATCH] SIENTIAPDE-1222 Enhance debug logging in MLFlowRepository to improve data traceability - Added a debug logging statement to capture received data for model prediction, improving visibility into input data. - Changed an existing info logging statement to debug level for consistency, ensuring all relevant data is logged at the appropriate level. --- laborious/utils/repository/model_repository.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 9d1cf64..18b4620 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -84,6 +84,9 @@ class MLFlowRepository(): input_index = data.index start_time = datetime.now() + + self.logger.debug( + f"Data received for model prediction: {data.to_string()}") data = self.model_serving.get_cached_predict( model_name, data, model_retention, flavor, compressed, retention_target @@ -91,7 +94,7 @@ class MLFlowRepository(): end_time = datetime.now() data = pd.DataFrame(data, columns=['prediction']) - self.logger.info( + self.logger.debug( f"Data received from model prediction: {data.to_string()}") data.index = input_index data['response_time'] = (end_time - start_time).total_seconds()