set dataset_location as the default location. (#1399)

2019-05-31 19:58:19 -07:00 · 2019-05-31 19:58:19 -07:00 · 88bca8223d
parent 87b96a0a7a
commit 88bca8223d
1 changed files with 1 additions and 1 deletions
--- 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