Update test_tokenization_auto.py

2020-01-13 22:38:29 -05:00
parent afc24ea5d4
commit 5cb463a714
1 changed files with 1 additions and 1 deletions
--- a/tests/test_tokenization_auto.py
+++ b/tests/test_tokenization_auto.py
@@ -33,7 +33,7 @@ class AutoTokenizerTest(unittest.TestCase):
    # @slow
    def test_tokenizer_from_pretrained(self):
        logging.basicConfig(level=logging.INFO)
-        for model_name in list(BERT_PRETRAINED_CONFIG_ARCHIVE_MAP.keys())[:1]:
+        for model_name in ["bert-base-uncased"]:
            tokenizer = AutoTokenizer.from_pretrained(model_name)
            self.assertIsNotNone(tokenizer)
            self.assertIsInstance(tokenizer, BertTokenizer)