diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index bf010c82aaf..fbea3dc5c3e 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery.py +++ b/sdk/python/feast/infra/offline_stores/bigquery.py @@ -608,8 +608,11 @@ def _wait_until_done(bq_job): client.cancel_job(bq_job.job_id) raise BigQueryJobCancelled(job_id=bq_job.job_id) - if bq_job.exception(): - raise bq_job.exception() + # We explicitly set the timeout to None because `google-api-core` changed the default value and + # breaks downstream libraries. + # https://github.com/googleapis/python-api-core/issues/479 + if bq_job.exception(timeout=None): + raise bq_job.exception(timeout=None) def _get_table_reference_for_new_entity(