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