From 5ad1ccb2979544d51b8d8f1e6b977e8ee9c73234 Mon Sep 17 00:00:00 2001 From: Akhilesh Halageri Date: Thu, 19 Oct 2023 21:17:45 +0000 Subject: [PATCH] fix: check for None; for when job is being created --- zetta_utils/cloud_management/resource_allocation/k8s/job.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/zetta_utils/cloud_management/resource_allocation/k8s/job.py b/zetta_utils/cloud_management/resource_allocation/k8s/job.py index af0e22a5b..152c5414e 100644 --- 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.")