fix: fix a minor bug (#694)

2026-07-28 16:07:46 +00:00 · 2025-03-17 21:43:26 +08:00
parent 93b41c1414
commit b8a699a0df
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_name] = val_pred
-          test_preds_dict[model_name] = test_pred
+          val_preds_dict[model_module] = val_pred
+          test_preds_dict[model_module] = test_pred
          {% endfor %}
          final_pred = ensemble_workflow(test_preds_dict, val_preds_dict, val_y)
          {% endraw %}