From 5cb463a714b0cc275388337d252dca76513c47d4 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Mon, 13 Jan 2020 22:38:29 -0500 Subject: [PATCH] Update test_tokenization_auto.py --- tests/test_tokenization_auto.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_tokenization_auto.py b/tests/test_tokenization_auto.py index 75849216bd..7691217779 100644 --- 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)