fix preds_dict's keys (#695)

2026-07-28 07:57:44 +00:00 · 2025-03-17 23:33:17 +08:00
parent 3d7b2b4567
commit 1d6e97cb99
1 changed files with 2 additions and 2 deletions
@@ -294,8 +294,8 @@ spec:
            val_y=val_y,
            test_X=X_test_transformed
          )
-          val_preds_dict[model_module] = val_pred
-          test_preds_dict[model_module] = test_pred
+          val_preds_dict[model_module.__name__] = val_pred
+          test_preds_dict[model_module.__name__] = test_pred
          {% endfor %}
          final_pred = ensemble_workflow(test_preds_dict, val_preds_dict, val_y)
          {% endraw %}