Skip to content

Commit 99e886b

Browse files
authored
Merge pull request #18 from voidful/codex/rename-testtok-method-and-run-tests
Rename roberta tokenization test
2 parents 48e91b4 + 60cb750 commit 99e886b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tfkit/test/utility/test_utility_tok.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def testTok(self):
2424
pad = tfkit.utility.tok.tok_pad(tokenizer)
2525
self.assertEqual(pad, "[PAD]")
2626

27-
def testTok(self):
27+
def testTok_roberta(self):
2828
tokenizer = AutoTokenizer.from_pretrained('distilroberta-base')
2929
begin = tfkit.utility.tok.tok_begin(tokenizer)
3030
self.assertEqual(begin, "<s>")

0 commit comments

Comments
 (0)