gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-29 08:27:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: duplicate model names test in pipeline coder & runner (#763)

Browse Source 
* add model name duplicate test in pipeline costeer

* fix ci
This commit is contained in:
XianBW
2025-04-07 19:06:42 +08:00
committed by GitHub
parent 4c5020f67e
commit e7f3cf97ec
3 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/pipeline/eval.py
+7 -4
View File
@@ -71,11 +71,14 @@ class PipelineCoSTEEREvaluator(CoSTEEREvaluator):
model_set_in_scores = set(score_df.index)
# Check model names (index)
if "ensemble" not in model_set_in_scores:
score_check_text += (
f"\n[Error] The score dataframe doesn't contain the ensemble model.\nscore_df is:\n{score_df}"
)
if not score_df.index.is_unique:
score_check_text += "\n[Error] The score dataframe contains duplicate model names."
score_ret_code = 1
if "ensemble" not in model_set_in_scores:
score_check_text += "\n[Error] The score dataframe doesn't contain the ensemble model."
score_ret_code = 1
if score_ret_code != 0:
score_check_text += f"The score_df is:\n{score_df}"
# Check metric name (columns)
if score_df.columns.tolist() != [self.scen.metric_name]: