fix: check for None; for when job is being created

ZettaAI · Oct 19, 2023 · 5ad1ccb · 5ad1ccb
1 parent 6fc169f
commit 5ad1ccb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/zetta_utils/cloud_management/resource_allocation/k8s/job.py b/zetta_utils/cloud_management/resource_allocation/k8s/job.py
@@ -162,13 +162,15 @@ def wait_for_job_completion(
         name=job.metadata.name,
         namespace=namespace,
     )
-    while job.status.succeeded == 0:
+    not_done = job.status.succeeded == 0 or job.status.succeeded is None
+    while not_done:
+        logger.info(f"Waiting for `{job.metadata.name}` to complete.")
         time.sleep(5)
         job = batch_v1_api.read_namespaced_job_status(
             name=job.metadata.name,
             namespace=namespace,
         )
-        logger.info(f"Waiting for `{job.metadata.name}` to complete.")
+        not_done = job.status.succeeded == 0 or job.status.succeeded is None
     logger.info(f"`{job.metadata.name}` job completed.")