From 1f5a14be3fdafd3dea60be238eff2f0a80ee3147 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 25 Mar 2026 11:18:21 -0300 Subject: [PATCH] SIENTIAPDE-1712 Add debug logging for non-DataFrame prediction data in MLFlowRepository --- laborious/utils/repository/model_repository.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index b5f9e9b..fc965dd 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -1283,6 +1283,10 @@ class MLFlowRepository(SientiaMonitoring): predict_data.columns = pd.Index(['prediction']) else: + self.debug( + f'Data received from model prediction (not a DataFrame): {predict_data}', + metadata, + ) predict_data = pd.DataFrame(predict_data, columns=['prediction']) # predict_data.to_csv( # f"tmp/predicted_data_{model_name}.csv", index=True)