From 80800dfdd208c8200e793b1c799e8825865b8108 Mon Sep 17 00:00:00 2001 From: Dery Rahman Ahaddienata Date: Wed, 24 Apr 2024 11:29:24 +0700 Subject: [PATCH] fix: assigned retry object --- task/bq2bq/executor/bumblebee/bigquery_service.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/task/bq2bq/executor/bumblebee/bigquery_service.py b/task/bq2bq/executor/bumblebee/bigquery_service.py index ccf767b..d22ee15 100644 --- 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