Unverified Commit a6938c47 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Patch BigBird tokenization test (#12653)

parent c523b241
...@@ -220,7 +220,7 @@ class BigBirdTokenizationTest(TokenizerTesterMixin, unittest.TestCase): ...@@ -220,7 +220,7 @@ class BigBirdTokenizationTest(TokenizerTesterMixin, unittest.TestCase):
tokenizer = BigBirdTokenizer.from_pretrained("google/bigbird-roberta-base") tokenizer = BigBirdTokenizer.from_pretrained("google/bigbird-roberta-base")
decoded_text = tokenizer.decode(tokenizer("Paris is the [MASK].").input_ids) decoded_text = tokenizer.decode(tokenizer("Paris is the [MASK].").input_ids)
self.assertTrue(decoded_text == "[CLS] Paris is the [MASK].[SEP]") self.assertTrue(decoded_text == "[CLS] Paris is the[MASK].[SEP]")
@slow @slow
def test_tokenizer_integration(self): def test_tokenizer_integration(self):
......
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