Skip to content

Commit

Permalink
fix: Ensure BigQuery queries are executed in UTC (#1174)
Browse files Browse the repository at this point in the history
  • Loading branch information
nj1973 committed Jun 21, 2024
1 parent 1879beb commit a77cdd9
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions data_validation/clients.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,12 +75,20 @@ def get_client_call(*args, **kwargs):

def get_bigquery_client(project_id, dataset_id="", credentials=None):
info = client_info.get_http_client_info()
job_config = bigquery.QueryJobConfig(
connection_properties=[bigquery.ConnectionProperty("time_zone", "UTC")]
)
google_client = bigquery.Client(
project=project_id, client_info=info, credentials=credentials
project=project_id,
client_info=info,
credentials=credentials,
default_query_job_config=job_config,
)

ibis_client = ibis.bigquery.connect(
project_id=project_id, dataset_id=dataset_id, credentials=credentials
project_id=project_id,
dataset_id=dataset_id,
credentials=credentials,
)

# Override the BigQuery client object to ensure the correct user agent is
Expand Down

0 comments on commit a77cdd9

Please sign in to comment.