From b63beb743c84b92d7fc416539161c96a4e2544c9 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Tue, 3 Nov 2020 09:49:29 -0500 Subject: [PATCH] Skip tatoeba tests if Tatoeba-Challenge not cloned (#8260) --- examples/seq2seq/test_tatoeba_conversion.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/seq2seq/test_tatoeba_conversion.py b/examples/seq2seq/test_tatoeba_conversion.py index f344be0b00..6c08df0a63 100644 --- a/examples/seq2seq/test_tatoeba_conversion.py +++ b/examples/seq2seq/test_tatoeba_conversion.py @@ -1,11 +1,13 @@ +import os import tempfile import unittest -from transformers.convert_marian_tatoeba_to_pytorch import TatoebaConverter +from transformers.convert_marian_tatoeba_to_pytorch import DEFAULT_REPO, TatoebaConverter from transformers.file_utils import cached_property from transformers.testing_utils import slow +@unittest.skipUnless(os.path.exists(DEFAULT_REPO)) class TatoebaConversionTester(unittest.TestCase): @cached_property def resolver(self):