fix single node auto eval

allenai · Oct 4, 2024 · 28e20cb · 28e20cb
1 parent 83f6e2b
commit 28e20cb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/open_instruct/utils.py b/open_instruct/utils.py
@@ -713,7 +713,10 @@ def get_beaker_experiment_info(experiment_id: str) -> Optional[dict]:
 
 def beaker_experiment_succeeded(experiment_id: str) -> bool:
     experiment = get_beaker_experiment_info(experiment_id)
-    num_replicas = experiment["jobs"][0]["execution"]["spec"]["replicas"]
+    if "replicas" in experiment["jobs"][0]["execution"]["spec"]:
+        num_replicas = experiment["jobs"][0]["execution"]["spec"]["replicas"]
+    else:
+        num_replicas = 1
     if not experiment:
         return False
     pprint(experiment)