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

feat: add model removal and adjust some framework logic (#681)

Browse Source 
* prune model task

* add component_description

* add model removal logic to component, hypo, and task gen

* fix ci

* adjust coder to meet the requirement of model removal

* fix and refine the logic of model removal

* add model removal logic in model_eval

* fix ci

* fix ci

* prune some unnecessary codes
This commit is contained in:
Yuante Li
2025-03-18 14:42:53 +08:00
committed by GitHub
parent 229866123b
commit 3e9ba73c2f
9 changed files with 137 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/model/__init__.py
+4
View File
@@ -103,6 +103,10 @@ class ModelMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
f"{target_task.name}.py"
] != workspace.file_dict.get(f"{target_task.name}.py"):
break
# If the task involves model removal, assume it can only process one model at a time.
if len(batch_edit) == 1 and batch_edit[f"{target_task.name}.py"] == "__DEL__":
break
else:
raise CoderError("Failed to generate a new model code.")