Skip to content

Commit 4377db8

Browse files
committed
Update task_adapter.py
1 parent f6a715b commit 4377db8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/algotune/task_adapter.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -616,7 +616,7 @@ def measure_baseline_performance(task_instance, problem, num_runs=3, warmup_runs
616616
for _ in range(num_runs):
617617
start_time = time.perf_counter()
618618
try:
619-
result = task_instance.solve(problem)
619+
result = run_with_timeout(task_instance.solve, args=(problem,), timeout_seconds=30)
620620
end_time = time.perf_counter()
621621
if result is not None:
622622
elapsed_ms = (end_time - start_time) * 1000
@@ -745,7 +745,7 @@ def evaluate(program_path, config=None):
745745
config = {{
746746
"algotune": {{
747747
"num_trials": 5,
748-
"data_size": 5,
748+
"data_size": 500,
749749
"timeout": 30,
750750
"num_runs": 3,
751751
"warmup_runs": 1
@@ -961,7 +961,7 @@ def evaluate_stage1(program_path, config=None):
961961
config = {{
962962
"algotune": {{
963963
"num_trials": 5,
964-
"data_size": 5,
964+
"data_size": 500,
965965
"timeout": 30
966966
}}
967967
}}
@@ -1166,7 +1166,7 @@ def replace_latex_command(match):
11661166
# AlgoTune task-specific configuration
11671167
algotune:
11681168
num_trials: 5
1169-
data_size: 5
1169+
data_size: 500
11701170
timeout: 30
11711171
num_runs: 3
11721172
warmup_runs: 1

0 commit comments

Comments
 (0)