Added back support for preemptible setting

broadinstitute · Nov 3, 2023 · 075d697 · 075d697
1 parent fdbaade
commit 075d697
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/cli/sparklespray/config.py b/cli/sparklespray/config.py
@@ -243,9 +243,16 @@ def consume(name: str, default : Union[NoDefault, T]=NO_DEFAULT, parser : Callab
         ),
     )
 
-    config.max_preemptable_attempts_scale = consume(
-        "max_preemptable_attempts_scale", 2, int
-    )
+    preemptible_yn = consume("preemptible", "y")
+    assert preemptible_yn.lower() in ['y', 'n'], f"expected preemptible should be either y or n but value was: {preemptible_yn}"
+    if preemptible_yn.lower() == "n":
+        assert "max_preemptable_attempts_scale" not in config_dict, f"Cannot specify both preemptible=n and max_preemptable_attempts_scale"
+
+        config.max_preemptable_attempts_scale = 0
+    else:
+        config.max_preemptable_attempts_scale = consume(
+            "max_preemptable_attempts_scale", 2, int
+        )
 
     #    config.
     #    allowed_parameters = set(