From b5ae79e642f5a029e2a1502637ce45db08718aca Mon Sep 17 00:00:00 2001 From: John Giorgi Date: Wed, 15 Jun 2022 09:30:37 -0700 Subject: [PATCH] Fix typos in back_translation module --- nlpaug/augmenter/word/back_translation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nlpaug/augmenter/word/back_translation.py b/nlpaug/augmenter/word/back_translation.py index 2c391ef..9202f6f 100644 --- a/nlpaug/augmenter/word/back_translation.py +++ b/nlpaug/augmenter/word/back_translation.py @@ -11,7 +11,7 @@ BACK_TRANSLATION_MODELS = {} -def init_back_translatoin_model(from_model_name, to_model_name, device, force_reload=False, +def init_back_translation_model(from_model_name, to_model_name, device, force_reload=False, batch_size=32, max_length=None): global BACK_TRANSLATION_MODELS @@ -73,7 +73,7 @@ def substitute(self, data, n=1): @classmethod def get_model(cls, from_model_name, to_model_name, device='cuda', force_reload=False, batch_size=32, max_length=None): - return init_back_translatoin_model(from_model_name, to_model_name, device, + return init_back_translation_model(from_model_name, to_model_name, device, force_reload, batch_size, max_length) @classmethod