gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-09 21:10:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf: some small upgrade to factor costeer to improve the performance (#420)

Browse Source 
* 1. use dataframe.info instead of head
2. in former trace query, add the latest attempt to the last success execution

* fix CI
This commit is contained in:
Xu Yang
2024-10-10 20:29:17 +08:00
committed by GitHub
parent 5402c00165
commit a6ddf08e2a
6 changed files with 44 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/factor_coder/CoSTEER/scheduler.py
+1 -1
View File
@@ -40,7 +40,7 @@ def LLMSelect(
# find corresponding former trace for each task
target_factor_task_information = evo.sub_tasks[i].get_task_information()
if target_factor_task_information in former_trace:
tasks.append((i, evo.sub_tasks[i], former_trace[target_factor_task_information]))
tasks.append((i, evo.sub_tasks[i], former_trace[target_factor_task_information][0]))
system_prompt = (
Environment(undefined=StrictUndefined)