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)