From a4419af47a0b82c808771b8881f975015ba43f69 Mon Sep 17 00:00:00 2001
From: Matt <rocketknight1@gmail.com>
Date: Wed, 18 Oct 2023 13:34:59 +0100
Subject: [PATCH] Mark test_add_special_tokens as slow for whisper because it
 causes CI timeouts

---
 tests/models/whisper/test_tokenization_whisper.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/models/whisper/test_tokenization_whisper.py b/tests/models/whisper/test_tokenization_whisper.py
index fd1c135deb53..cbd8f0df72f9 100644
--- 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"