Mark test_add_special_tokens as slow for whisper because it causes CI…

… timeouts
huggingface · Oct 18, 2023 · a4419af · a4419af
1 parent 34678db
commit a4419af
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/models/whisper/test_tokenization_whisper.py b/tests/models/whisper/test_tokenization_whisper.py
@@ -307,6 +307,11 @@ def test_basic_normalizer(self):
         )
         self.assertEqual(decoded_output_diacritics, expected_output_diacritics)
 
+    @slow
+    def test_add_special_tokens(self):
+        # This test takes a long time for Whisper, so it's marked as slow for this model
+        super().test_add_special_tokens()
+
 
 class SpeechToTextTokenizerMultilinguialTest(unittest.TestCase):
     checkpoint_name = "openai/whisper-small.en"