fixed flag param for autorestart

valohai · Aug 31, 2022 · 40b57a2 · 40b57a2
1 parent 28516e0
commit 40b57a2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/valohai_cli/commands/execution/run/frontend_command.py b/valohai_cli/commands/execution/run/frontend_command.py
@@ -111,8 +111,8 @@ def run(
                     f"does not seem valid (it should start with `ssh`)"
                 )
         runtime_config["debug_key"] = key
-    if autorestart_spot:
-        runtime_config["autorestart"] = autorestart_spot
+    if autorestart:
+        runtime_config["autorestart"] = autorestart
 
     rc = RunCommand(
         project=project,