Minor fixes

microsoft · Aug 26, 2024 · 16b373f · 16b373f
1 parent f3fee23
commit 16b373f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlos_bench/mlos_bench/services/remote/azure/azure_vm_services.py b/mlos_bench/mlos_bench/services/remote/azure/azure_vm_services.py
@@ -618,7 +618,7 @@ def _check_remote_exec_status(self, params: dict) -> Tuple[Status, dict]:
             execution_state = (
                 output.get("properties", {}).get("instanceView", {}).get("executionState")
             )
-            if execution_state in {"Running", "Pending"}:
+            if execution_state in {"Running", "Pending", "Unknown"}:
                 return Status.RUNNING, {}
             elif execution_state == "Succeeded":
                 return Status.SUCCEEDED, output