[fix] Marian tests import (#5442)
This commit is contained in:
Родитель
608d5a7c44
Коммит
e0d58ddb65
|
@ -20,11 +20,11 @@ import unittest
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from shutil import copyfile
|
from shutil import copyfile
|
||||||
|
|
||||||
|
from transformers.testing_utils import _torch_available
|
||||||
from transformers.tokenization_marian import MarianTokenizer, save_json, vocab_files_names
|
from transformers.tokenization_marian import MarianTokenizer, save_json, vocab_files_names
|
||||||
from transformers.tokenization_utils import BatchEncoding
|
from transformers.tokenization_utils import BatchEncoding
|
||||||
|
|
||||||
from .test_tokenization_common import TokenizerTesterMixin
|
from .test_tokenization_common import TokenizerTesterMixin
|
||||||
from .utils import _torch_available
|
|
||||||
|
|
||||||
|
|
||||||
SAMPLE_SP = os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures/test_sentencepiece.model")
|
SAMPLE_SP = os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures/test_sentencepiece.model")
|
||||||
|
|
Загрузка…
Ссылка в новой задаче