From 9889d9f1b76f6173b81d8a64a5f382f4f1bf4823 Mon Sep 17 00:00:00 2001 From: mshannon-sil Date: Thu, 7 Nov 2024 15:59:39 -0500 Subject: [PATCH] fix test import --- .../huggingface/test_hugging_face_nmt_model_trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/translation/huggingface/test_hugging_face_nmt_model_trainer.py b/tests/translation/huggingface/test_hugging_face_nmt_model_trainer.py index 4094616..123d2a0 100644 --- a/tests/translation/huggingface/test_hugging_face_nmt_model_trainer.py +++ b/tests/translation/huggingface/test_hugging_face_nmt_model_trainer.py @@ -21,8 +21,7 @@ ) from machine.corpora import DictionaryTextCorpus, MemoryText, TextRow -from machine.translation.huggingface import HuggingFaceNmtEngine, HuggingFaceNmtModelTrainer -from machine.translation.huggingface.hugging_face_nmt_model_trainer import add_lang_code_to_tokenizer +from machine.translation.huggingface import HuggingFaceNmtEngine, HuggingFaceNmtModelTrainer, add_lang_code_to_tokenizer def test_train_non_empty_corpus() -> None: