gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-04 02:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update code to fix a small bug in model cache md5 hash (#303)

Browse Source 
* update code to fix a small bug in model cache md5 hash

* fix another bug dumping the wrong name to costeer model

* fix a black CI
This commit is contained in:
Xu Yang
2024-09-23 20:27:34 +08:00
committed by GitHub
parent 444ded9a5d
commit 9bb214fff2
2 changed files with 6 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/model_coder/CoSTEER/evolving_strategy.py
+2 -14
View File
@@ -146,21 +146,9 @@ class ModelCoderEvolvingStrategy(EvolvingStrategy):
n=RD_AGENT_SETTINGS.multi_proc_n,
)
model_file_mapping = {
"XGBoost": "model_xgb.py",
"RandomForest": "model_rf.py",
"LightGBM": "model_lgb.py",
"NN": "model_nn.py",
}
for index, target_index in enumerate(to_be_finished_task_index):
if evo.sub_workspace_list[target_index] is None:
evo.sub_workspace_list[target_index] = ModelFBWorkspace(target_task=evo.sub_tasks[target_index])
model_type = evo.sub_tasks[target_index].model_type
if model_type in model_file_mapping:
evo.sub_workspace_list[target_index].inject_code(**{model_file_mapping[model_type]: result[index]})
else:
raise ValueError(f"Unsupported model type: {model_type}")
evo.sub_workspace_list[target_index] = ModelFBWorkspace(target_task=evo.sub_tasks[target_index])
evo.sub_workspace_list[target_index].inject_code(**{"model.py": result[index]})
evo.corresponding_selection = to_be_finished_task_index