gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1712

Browse Source 
Add debug logging for non-DataFrame prediction data in MLFlowRepository
This commit is contained in:
vitor-aignosi
2026-03-25 11:18:21 -03:00
parent 367b00f00a
commit 1f5a14be3f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
laborious/utils/repository/model_repository.py
View File

@@ -1283,6 +1283,10 @@ class MLFlowRepository(SientiaMonitoring):
predict_data.columns = pd.Index(['prediction']) predict_data.columns = pd.Index(['prediction'])
else: 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 = pd.DataFrame(predict_data, columns=['prediction'])
# predict_data.to_csv( # predict_data.to_csv(
# f"tmp/predicted_data_{model_name}.csv", index=True) # f"tmp/predicted_data_{model_name}.csv", index=True)