fix: assigned retry object

goto · Apr 24, 2024 · 80800df · 80800df
1 parent 9dea8ca
commit 80800df
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/task/bq2bq/executor/bumblebee/bigquery_service.py b/task/bq2bq/executor/bumblebee/bigquery_service.py
@@ -73,8 +73,7 @@ def __init__(self, client, labels, writer, retry_timeout = None, on_job_finish =
         )
         predicate = if_exception_funcs(if_transient_error, if_additional_transient_error)
         retry = bigquery.DEFAULT_RETRY.with_deadline(retry_timeout) if retry_timeout else bigquery.DEFAULT_RETRY
-        retry.with_predicate(predicate)
-        self.retry = retry
+        self.retry = retry.with_predicate(predicate)
         self.on_job_finish = on_job_finish
         self.on_job_register = on_job_register