HGI-6352: Add backoff for ServiceUnavailable exception

Fix backoff declaration
hotgluexyz · Aug 14, 2024 · 0bbbce2 · 0bbbce2
1 parent fbb8d4b
commit 0bbbce2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/target_bigquery/processhandler.py b/target_bigquery/processhandler.py
@@ -256,7 +256,10 @@ def primary_key_condition(self, stream):
 
     @backoff.on_exception(
         backoff.expo,
-        google_exceptions.BadRequest,
+        (
+            google_exceptions.BadRequest,
+            google_exceptions.ServiceUnavailable,
+        ),
         max_tries=3
     )
     def _do_temp_table_based_load(self, rows):