fix: fix a bug related to model_name in ensemble (#692)

* fix a bug related to model_name in ensemble * refine
2026-08-05 19:17:43 +00:00 · 2025-03-17 21:23:57 +08:00
parent e0ae91b497
commit bb91201dd2
2 changed files with 12 additions and 9 deletions
@@ -81,6 +81,7 @@ class WorkflowGeneralCaseSpecEvaluator(CoSTEEREvaluator):
            try:
                score_df = pd.read_csv(score_fp, index_col=0)
                model_set_in_scores = set(score_df.index)
+                # We assume that model names in `score_df` are stored without the '.py' file extension.
                model_set_in_folder = set(
                    f[:-3] for f in implementation.file_dict.keys() if re.match(r"^model_(?!test)\w+\.py$", f)
                )