[ASR Pipe Test] Fix CTC timestamps error message (huggingface#25727)

hotchpotch · Aug 24, 2023 · 0218876 · 0218876
1 parent fd0b94f
commit 0218876
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/transformers/pipelines/automatic_speech_recognition.py b/src/transformers/pipelines/automatic_speech_recognition.py
@@ -402,7 +402,7 @@ def _sanitize_parameters(
                 raise ValueError("CTC with LM can only predict word level timestamps, set `return_timestamps='word'`")
             if self.type == "ctc" and return_timestamps not in ["char", "word"]:
                 raise ValueError(
-                    "CTC can either predict character (char) level timestamps, or word level timestamps."
+                    "CTC can either predict character level timestamps, or word level timestamps."
                     "Set `return_timestamps='char'` or `return_timestamps='word'` as required."
                 )
             if self.type == "seq2seq_whisper" and return_timestamps == "char":

diff --git a/tests/pipelines/test_pipelines_automatic_speech_recognition.py b/tests/pipelines/test_pipelines_automatic_speech_recognition.py
@@ -1150,7 +1150,7 @@ def test_chunking_and_timestamps(self):
         # CTC models must specify return_timestamps type - cannot set `return_timestamps=True` blindly
         with self.assertRaisesRegex(
             ValueError,
-            "^CTC can either predict character (char) level timestamps, or word level timestamps."
+            "^CTC can either predict character level timestamps, or word level timestamps."
             "Set `return_timestamps='char'` or `return_timestamps='word'` as required.$",
         ):
             _ = speech_recognizer(audio, return_timestamps=True)