HGI-6352: Add backoff for ServiceUnavailable exception

hotgluexyz · Aug 14, 2024 · 59572b9 · 59572b9
1 parent fbb8d4b
commit 59572b9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/target_bigquery/processhandler.py b/target_bigquery/processhandler.py
@@ -257,6 +257,7 @@ def primary_key_condition(self, stream):
     @backoff.on_exception(
         backoff.expo,
         google_exceptions.BadRequest,
+        google_exceptions.ServiceUnavailable,
         max_tries=3
     )
     def _do_temp_table_based_load(self, rows):