From e962a08f356a3a9b3585ed5ce9a5da13dd9fbd2c Mon Sep 17 00:00:00 2001 From: anenin <127075025+anenin23@users.noreply.github.com> Date: Wed, 6 Nov 2024 18:09:09 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20Fix=20parameter=20mismatch=20in=20EPUBBo?= =?UTF-8?q?okLoaderHelper.translate=5Fwith=5Fba=E2=80=A6=20(#429)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * fix: Fix parameter mismatch in EPUBBookLoaderHelper.translate_with_backoff - Fix TypeError when calling translate_with_backoff with multiple arguments - Add proper parameter handling in the decorated method - Add jitter=None to prevent extra parameters from backoff decorator - Improve code readability and error handling * style: format code with black --------- Co-authored-by: wenping --- book_maker/loader/helper.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/book_maker/loader/helper.py b/book_maker/loader/helper.py index b628f4f..ba5485e 100644 --- a/book_maker/loader/helper.py +++ b/book_maker/loader/helper.py @@ -1,7 +1,7 @@ import re -from copy import copy import backoff import logging +from copy import copy logging.basicConfig(level=logging.WARNING) logger = logging.getLogger(__name__) @@ -37,9 +37,10 @@ class EPUBBookLoaderHelper: Exception, on_backoff=lambda details: logger.warning(f"retry backoff: {details}"), on_giveup=lambda details: logger.warning(f"retry abort: {details}"), + jitter=None, ) - def translate_with_backoff(self, **kwargs): - return self.translate_model.translate(**kwargs) + def translate_with_backoff(self, text, context_flag=False): + return self.translate_model.translate(text, context_flag) def deal_new(self, p, wait_p_list, single_translate=False): self.deal_old(wait_p_list, single_translate, self.context_flag)