Increase materialization job timeout

vinted · Mar 5, 2024 · 82a68f2 · 82a68f2
1 parent 0ae8b14
commit 82a68f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -714,7 +714,7 @@ def _upload_entity_df(
     else:
         raise InvalidEntityType(type(entity_df))
 
-    block_until_done(client, job)
+    block_until_done(client, job, timeout=3600)
 
     # Ensure that the table expires after some time
     table = client.get_table(table=table_name)