From f39785fc130de5e48db8ae63ccdf34de91a42b41 Mon Sep 17 00:00:00 2001 From: Igor Dvorzhak Date: Mon, 20 Dec 2021 18:59:35 -0800 Subject: [PATCH 1/2] Fix job polling --- samples/snippets/submit_job_to_cluster.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/samples/snippets/submit_job_to_cluster.py b/samples/snippets/submit_job_to_cluster.py index 93a9f66c..7536255b 100644 --- a/samples/snippets/submit_job_to_cluster.py +++ b/samples/snippets/submit_job_to_cluster.py @@ -202,9 +202,9 @@ def wait_for_job(dataproc, project, region, job_id): request={"project_id": project, "region": region, "job_id": job_id} ) # Handle exceptions - if job.status.State.Name(job.status.state) == "ERROR": + if job.status.State(job.status.state).name == "ERROR": raise Exception(job.status.details) - elif job.status.State.Name(job.status.state) == "DONE": + elif job.status.State(job.status.state).name == "DONE": print("Job finished.") return job From 1ef192702f289b7febe54b60199d123ae8014ad7 Mon Sep 17 00:00:00 2001 From: Igor Dvorzhak Date: Wed, 22 Dec 2021 15:02:37 -0800 Subject: [PATCH 2/2] Update submit_job_to_cluster.py --- samples/snippets/submit_job_to_cluster.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samples/snippets/submit_job_to_cluster.py b/samples/snippets/submit_job_to_cluster.py index 7536255b..b1024675 100644 --- a/samples/snippets/submit_job_to_cluster.py +++ b/samples/snippets/submit_job_to_cluster.py @@ -204,7 +204,7 @@ def wait_for_job(dataproc, project, region, job_id): # Handle exceptions if job.status.State(job.status.state).name == "ERROR": raise Exception(job.status.details) - elif job.status.State(job.status.state).name == "DONE": + if job.status.State(job.status.state).name == "DONE": print("Job finished.") return job