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

fix: increase time default not controlled by LLM (#1196)

Browse Source 
* fix: add longer timeout configuration for data science scenarios

* fix CI
This commit is contained in:
Xu Yang
2025-08-21 11:27:17 +08:00
committed by GitHub
parent 04878f9e70
commit e4bd647d1b
2 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/app/data_science/conf.py
+1
View File
@@ -140,6 +140,7 @@ class DataScienceBasePropSetting(KaggleBasePropSetting):
model_architecture_suggestion_time_percent: float = 0.75 model_architecture_suggestion_time_percent: float = 0.75
allow_longer_timeout: bool = False allow_longer_timeout: bool = False
longer_timeout_by_llm: bool = False
coder_longer_timeout_multiplier_upper: int = 3 coder_longer_timeout_multiplier_upper: int = 3
runner_longer_timeout_multiplier_upper: int = 2 runner_longer_timeout_multiplier_upper: int = 2
timeout_increase_stage: float = 0.3 timeout_increase_stage: float = 0.3
rdagent/scenarios/data_science/scen/__init__.py
+13 -5
View File
@@ -119,9 +119,17 @@ class DataScienceScen(Scenario):
) )
self.metric_name = response_json_analysis.get("Metric Name", "custom_metric") self.metric_name = response_json_analysis.get("Metric Name", "custom_metric")
self.metric_direction_guess = response_json_analysis.get("Metric Direction", True) self.metric_direction_guess = response_json_analysis.get("Metric Direction", True)
self.longer_time_limit_required = response_json_analysis.get( self.longer_time_limit_required = (
"Longer time limit required", False False
) # True or False, whether the competition scenario requires a longer time limit to the code. if not DS_RD_SETTING.allow_longer_timeout
else (
response_json_analysis.get("Longer time limit required", False)
if DS_RD_SETTING.longer_timeout_by_llm
else True
)
)
# True or False, whether the competition scenario requires a longer time limit to the code.
def real_debug_timeout(self): def real_debug_timeout(self):
return ( return (
@@ -130,7 +138,7 @@ class DataScienceScen(Scenario):
DS_RD_SETTING.coder_longer_timeout_multiplier_upper, DS_RD_SETTING.coder_longer_timeout_multiplier_upper,
self.timeout_increase_count * DS_RD_SETTING.timeout_increase_stage + 1, self.timeout_increase_count * DS_RD_SETTING.timeout_increase_stage + 1,
) )
if self.longer_time_limit_required and DS_RD_SETTING.allow_longer_timeout if self.longer_time_limit_required
else DS_RD_SETTING.debug_timeout else DS_RD_SETTING.debug_timeout
) )
@@ -144,7 +152,7 @@ class DataScienceScen(Scenario):
DS_RD_SETTING.runner_longer_timeout_multiplier_upper, DS_RD_SETTING.runner_longer_timeout_multiplier_upper,
self.timeout_increase_count * DS_RD_SETTING.timeout_increase_stage + 1, self.timeout_increase_count * DS_RD_SETTING.timeout_increase_stage + 1,
) )
if self.longer_time_limit_required and DS_RD_SETTING.allow_longer_timeout if self.longer_time_limit_required
else DS_RD_SETTING.full_timeout else DS_RD_SETTING.full_timeout
) )