Merge pull request #298 from JohnGiorgi/patch-1

Fix typos in back_translation module
makcedward · Jul 1, 2022 · 14cf962 · 14cf962
2 parents 38a7426 + b5ae79e
commit 14cf962
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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