relax network connection requirements

2019-04-17 14:22:35 +02:00
parent fa76520240
commit 265550ec34
2 changed files with 23 additions and 8 deletions
--- a/tests/tokenization_gpt2_test.py
+++ b/tests/tokenization_gpt2_test.py
@@ -66,7 +66,7 @@ class GPT2TokenizationTest(unittest.TestCase):
            [tokenizer_2.encoder, tokenizer_2.decoder, tokenizer_2.bpe_ranks,
             tokenizer_2.special_tokens, tokenizer_2.special_tokens_decoder])

-    @pytest.mark.slow
+    # @pytest.mark.slow
    def test_tokenizer_from_pretrained(self):
        cache_dir = "/tmp/pytorch_pretrained_bert_test/"
        for model_name in list(PRETRAINED_VOCAB_ARCHIVE_MAP.keys())[:1]: