Merge pull request #170 from harsha-simhadri/harshasi/timeout_fix

update time out rules for neurip23 tracks
harsha-simhadri · Oct 2, 2023 · 5a8d106 · 5a8d106
2 parents 3155952 + b37b457
commit 5a8d106
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/benchmark/runner.py b/benchmark/runner.py
@@ -290,8 +290,9 @@ def run_docker(definition, dataset, count, runs, timeout, rebuild,
 
     # set/override container timeout based on competition flag
     if neurips23track!='none':
-        timeout = 30*60 # 30 minutes
-        print("Setting container wait timeout to 30 minutes")       
+        # 1 hour for streaming and 12 hours for other tracks
+        timeout = 60 * 60 if neurips23track == 'streaming' else 12 * 60 * 60
+        print("Setting container wait timeout to %d seconds" % timeout)       
 
     elif not timeout: 
         # default to 3 days (includes NeurIPS'21)