debug info, probably dislib issue

eflows4hpc · Feb 5, 2024 · e1d517b · e1d517b
1 parent 384cf56
commit e1d517b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model_search_upload.py b/model_search_upload.py
@@ -82,6 +82,7 @@ def uploat_to_mlflow(temp_dir, **context):
         print("Uploading model search results")
         df = pd.read_csv(os.path.join(temp_dir, 'pd.csv'), index_col=0)
         dct = df.to_dict()
+        print("Got following data", dct)
         metrics=['mean_test_score', 'mean_fit_time']
 
         for i, p in enumerate(dct['params'].values()):