fix tests - flagged as slow all the tests downloading from AWS

This commit is contained in:
thomwolf
2019-11-04 16:03:36 +01:00
parent f02805da6f
commit b340a910ed
11 changed files with 21 additions and 7 deletions

View File

@@ -27,6 +27,7 @@ else:
class EncoderDecoderModelTest(unittest.TestCase):
@pytest.mark.slow
def test_model2model_from_pretrained(self):
logging.basicConfig(level=logging.INFO)
for model_name in list(BERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: