From 88bca8223da3b51e44d764e7a551eff32cad5d5a Mon Sep 17 00:00:00 2001
From: hongye-sun <43763191+hongye-sun@users.noreply.github.com>
Date: Fri, 31 May 2019 19:58:19 -0700
Subject: [PATCH] set dataset_location as the default location. (#1399)

---
 component_sdk/python/kfp_component/google/bigquery/_query.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/component_sdk/python/kfp_component/google/bigquery/_query.py b/component_sdk/python/kfp_component/google/bigquery/_query.py
index 9559dc058f..bf147d7b37 100644
--- a/component_sdk/python/kfp_component/google/bigquery/_query.py
+++ b/component_sdk/python/kfp_component/google/bigquery/_query.py
@@ -43,7 +43,7 @@ def query(query, project_id, dataset_id=None, table_id=None,
     Returns:
         The API representation of the completed query job.
     """
-    client = bigquery.Client(project=project_id)
+    client = bigquery.Client(project=project_id, location=dataset_location)
     if not job_config:
         job_config = bigquery.QueryJobConfig()
         job_config.create_disposition = bigquery.job.CreateDisposition.CREATE_IF_NEEDED