diff --git a/rdagent/scenarios/data_science/scen/__init__.py b/rdagent/scenarios/data_science/scen/__init__.py index 8ea68da7..fe2158b4 100644 --- a/rdagent/scenarios/data_science/scen/__init__.py +++ b/rdagent/scenarios/data_science/scen/__init__.py @@ -128,15 +128,7 @@ class DataScienceScen(Scenario): ) def recommend_debug_timeout(self): - return ( - DS_RD_SETTING.debug_recommend_timeout - * min( - DS_RD_SETTING.coder_longer_timeout_multiplier_upper, - self.timeout_increase_count * DS_RD_SETTING.timeout_increase_stage + 1, - ) - if self.longer_time_limit_required and DS_RD_SETTING.allow_longer_timeout - else DS_RD_SETTING.debug_timeout - ) + return DS_RD_SETTING.debug_recommend_timeout def real_full_timeout(self): return ( @@ -150,15 +142,7 @@ class DataScienceScen(Scenario): ) def recommend_full_timeout(self): - return ( - DS_RD_SETTING.full_recommend_timeout - * min( - DS_RD_SETTING.runner_longer_timeout_multiplier_upper, - self.timeout_increase_count * DS_RD_SETTING.timeout_increase_stage + 1, - ) - if self.longer_time_limit_required and DS_RD_SETTING.allow_longer_timeout - else DS_RD_SETTING.full_recommend_timeout - ) + return DS_RD_SETTING.full_recommend_timeout def increase_timeout(self): """Increase the timeout multiplier for the scenario."""