From e0d58ddb65eff1a52572dff75944d8b28ea706d3 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Wed, 1 Jul 2020 11:42:22 -0400 Subject: [PATCH] [fix] Marian tests import (#5442) --- tests/test_tokenization_marian.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_tokenization_marian.py b/tests/test_tokenization_marian.py index 8a189651aa..693314bfc4 100644 --- a/tests/test_tokenization_marian.py +++ b/tests/test_tokenization_marian.py @@ -20,11 +20,11 @@ import unittest from pathlib import Path 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_utils import BatchEncoding 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")