Pass location where auto-detection not supported

MaxxleLLC · Oct 25, 2019 · 04555f9 · 04555f9
1 parent 0e50ead
commit 04555f9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/bigquery/samples/client_query_batch.py b/bigquery/samples/client_query_batch.py
@@ -36,7 +36,9 @@ def client_query_batch(client):
 
     # Check on the progress by getting the job's updated state. Once the state
     # is `DONE`, the results are ready.
-    query_job = client.get_job(query_job.job_id)  # Make an API request.
+    query_job = client.get_job(
+        query_job.job_id, location=query_job.location
+    )  # Make an API request.
 
     print("Job {} is currently in state {}".format(query_job.job_id, query_job.state))
     # [END bigquery_query_batch]

diff --git a/bigquery/samples/tests/test_create_job.py b/bigquery/samples/tests/test_create_job.py
@@ -19,6 +19,6 @@
 def test_create_job(capsys, client):
 
     query_job = create_job.create_job(client)
-    client.cancel_job(query_job.job_id)
+    client.cancel_job(query_job.job_id, location=query_job.location)
     out, err = capsys.readouterr()
     assert "Started job: {}".format(query_job.job_id) in out