Unverified Commit d5831acb authored by Julien Chaumond's avatar Julien Chaumond Committed by GitHub
Browse files

Update test_tokenization_auto.py

parent ed6cd597
......@@ -33,7 +33,7 @@ class AutoTokenizerTest(unittest.TestCase):
# @slow
def test_tokenizer_from_pretrained(self):
logging.basicConfig(level=logging.INFO)
for model_name in [x for x in BERT_PRETRAINED_CONFIG_ARCHIVE_MAP if not "japanese" in x]:
for model_name in [x for x in BERT_PRETRAINED_CONFIG_ARCHIVE_MAP.keys() if not "japanese" in x]:
tokenizer = AutoTokenizer.from_pretrained(model_name)
self.assertIsNotNone(tokenizer)
self.assertIsInstance(tokenizer, BertTokenizer)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment