From 7604557e4470822754c5658a19d81aa2ae7de934 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 14 Oct 2021 09:07:51 -0400 Subject: [PATCH] Fix FNet tokenizer tests (#13995) --- tests/test_tokenization_fnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_tokenization_fnet.py b/tests/test_tokenization_fnet.py index 7a4e5c1932..ed4f11ecce 100644 --- a/tests/test_tokenization_fnet.py +++ b/tests/test_tokenization_fnet.py @@ -17,7 +17,7 @@ import os import unittest from transformers import FNetTokenizer, FNetTokenizerFast -from transformers.testing_utils import require_sentencepiece, require_tokenizers, slow +from transformers.testing_utils import require_sentencepiece, require_tokenizers, slow, tooslow from transformers.tokenization_utils import AddedToken from .test_tokenization_common import TokenizerTesterMixin @@ -172,7 +172,7 @@ class FNetTokenizationTest(TokenizerTesterMixin, unittest.TestCase): self.assertTrue(special_token_id in p_output) self.assertTrue(special_token_id in cr_output) - @slow + @tooslow def test_special_tokens_initialization_from_slow(self): for tokenizer, pretrained_name, kwargs in self.tokenizers_list: with self.subTest(f"{tokenizer.__class__.__name__} ({pretrained_name})"): @@ -439,5 +439,5 @@ class FNetTokenizationTest(TokenizerTesterMixin, unittest.TestCase): self.tokenizer_integration_test_util( expected_encoding=expected_encoding, model_name="google/fnet-base", - revision="58e0d1f96af163dc8d0a84a2fddf4bd403e4e802", + revision="34219a71ca20e280cc6000b89673a169c65d605c", )