Skip to content

Commit 0919762

Browse files
committed
Update task_adapter.py
1 parent 9b288fd commit 0919762

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

examples/algotune/task_adapter.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -827,7 +827,7 @@ def evaluate(program_path, config=None):
827827
"algotune": {{
828828
"num_trials": 5,
829829
"data_size": 100,
830-
"timeout": 30,
830+
"timeout": 3000,
831831
"num_runs": 3,
832832
"warmup_runs": 1
833833
}}
@@ -837,7 +837,7 @@ def evaluate(program_path, config=None):
837837
algotune_config = config.get("algotune", {{}})
838838
num_trials = algotune_config.get("num_trials", 5)
839839
data_size = algotune_config.get("data_size", 5)
840-
timeout_seconds = algotune_config.get("timeout", 30)
840+
timeout_seconds = algotune_config.get("timeout", 300)
841841
num_runs = algotune_config.get("num_runs", 3)
842842
warmup_runs = algotune_config.get("warmup_runs", 1)
843843
@@ -1043,13 +1043,13 @@ def evaluate_stage1(program_path, config=None):
10431043
"algotune": {{
10441044
"num_trials": 5,
10451045
"data_size": 100,
1046-
"timeout": 30
1046+
"timeout": 300
10471047
}}
10481048
}}
10491049
10501050
algotune_config = config.get("algotune", {{}})
10511051
data_size = algotune_config.get("data_size", 5)
1052-
timeout_seconds = algotune_config.get("timeout", 30)
1052+
timeout_seconds = algotune_config.get("timeout", 300)
10531053
10541054
# Load the program
10551055
spec = importlib.util.spec_from_file_location("program", program_path)
@@ -1248,7 +1248,7 @@ def replace_latex_command(match):
12481248
algotune:
12491249
num_trials: 5
12501250
data_size: 100
1251-
timeout: 30
1251+
timeout: 300
12521252
num_runs: 3
12531253
warmup_runs: 1
12541254
'''

0 commit comments

Comments
 (0)