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

gbmarc1 · web-flow · commit c0ca7e4638d5 · 2022-12-19T15:33:59.000-08:00
* Specifies timeout in exception polling

Signed-off-by: gbmarc1 &lt;marcantoine.belanger@shopify.com&gt;

* Update bigquery.py

Signed-off-by: gbmarc1 &lt;marcantoine.belanger@shopify.com&gt;

Signed-off-by: gbmarc1 &lt;marcantoine.belanger@shopify.com&gt;
diff --git 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(
