Skip tatoeba tests if Tatoeba-Challenge not cloned (#8260)

2020-11-03 09:49:29 -05:00
parent 9f1747f999
commit b63beb743c
1 changed files with 3 additions and 1 deletions
--- 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):