fix: Specifies timeout in exception polling (feast-dev#3398)

* Specifies timeout in exception polling

Signed-off-by: gbmarc1 <[email protected]>

* Update bigquery.py

Signed-off-by: gbmarc1 <[email protected]>

Signed-off-by: gbmarc1 <[email protected]>

Author: gbmarc1

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(