From d356b89f3c1906164e3a567ba25d0b7134c46d9c Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Tue, 18 Oct 2022 08:56:37 +0200 Subject: [PATCH] fix test whisper with new max length (#19668) --- .../pipelines/test_pipelines_automatic_speech_recognition.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/pipelines/test_pipelines_automatic_speech_recognition.py b/tests/pipelines/test_pipelines_automatic_speech_recognition.py index 2aacbccbd4e..5338f59aacb 100644 --- a/tests/pipelines/test_pipelines_automatic_speech_recognition.py +++ b/tests/pipelines/test_pipelines_automatic_speech_recognition.py @@ -339,7 +339,10 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase, metaclass=Pipel ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation") filename = ds[0]["file"] output = speech_recognizer(filename) - self.assertEqual(output, {"text": " Mr. Quilter is the apostle of the middle classes, and we are glad to"}) + self.assertEqual( + output, + {"text": " Mr. Quilter is the apostle of the middle classes, and we are glad to welcome his gospel."}, + ) @slow @require_torch