SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update test_tokenization_auto.py

Browse Source
This commit is contained in:
Julien Chaumond
2020-01-13 22:46:35 -05:00
committed by GitHub
parent 5cb463a714
commit ed6cd597cc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_tokenization_auto.py
View File

@@ -33,7 +33,7 @@ class AutoTokenizerTest(unittest.TestCase):
# @slow # @slow
def test_tokenizer_from_pretrained(self): def test_tokenizer_from_pretrained(self):
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
for model_name in ["bert-base-uncased"]: for model_name in [x for x in BERT_PRETRAINED_CONFIG_ARCHIVE_MAP if not "japanese" in x]:
tokenizer = AutoTokenizer.from_pretrained(model_name) tokenizer = AutoTokenizer.from_pretrained(model_name)
self.assertIsNotNone(tokenizer) self.assertIsNotNone(tokenizer)
self.assertIsInstance(tokenizer, BertTokenizer) self.assertIsInstance(tokenizer, BertTokenizer)