Skip to content

Commit

Permalink
fix: Fix parameter mismatch in EPUBBookLoaderHelper.translate_with_ba… (
Browse files Browse the repository at this point in the history
#429)

* 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 <[email protected]>
  • Loading branch information
anenin23 and angenpn authored Nov 6, 2024
1 parent 546fbd8 commit e962a08
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions book_maker/loader/helper.py
Original file line number Diff line number Diff line change
@@ -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__)
Expand Down Expand Up @@ -37,9 +37,10 @@ def insert_trans(self, p, text, translation_style="", single_translate=False):
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)
Expand Down

0 comments on commit e962a08

Please sign in to comment.