riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix AutomaticSpeechRecognitionPipelineTests.run_pipeline_test (#20597)

Browse Source 
* Remove assert exception not triggered

* Fix wrong expected exception string

* fix

* use assertRaisesRegex

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2022-12-06 15:48:49 +01:00 committed by GitHub
parent 6a707cf586
commit 9b14c1b6bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/pipelines/test_pipelines_automatic_speech_recognition.py
View File

@ -118,15 +118,9 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase, metaclass=Pipel
}, },
) )
else: else:
# Non CTC models cannot use chunk_length
with self.assertRaises(ValueError) as v:
outputs = speech_recognizer(audio, chunk_length_s=10)
self.assertEqual(v.exception, "")
# Non CTC models cannot use return_timestamps # Non CTC models cannot use return_timestamps
with self.assertRaises(ValueError) as v: with self.assertRaisesRegex(ValueError, "^We cannot return_timestamps yet on non-ctc models !$"):
outputs = speech_recognizer(audio, return_timestamps="char") outputs = speech_recognizer(audio, return_timestamps="char")
self.assertEqual(v.exception, "")
@require_torch @require_torch
@slow @slow
@ -148,9 +142,8 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase, metaclass=Pipel
self.assertEqual(output, {"text": "(Applaudissements)"}) self.assertEqual(output, {"text": "(Applaudissements)"})
# Non CTC models cannot use return_timestamps # Non CTC models cannot use return_timestamps
with self.assertRaises(ValueError) as v: with self.assertRaisesRegex(ValueError, "^We cannot return_timestamps yet on non-ctc models !$"):
_ = speech_recognizer(waveform, return_timestamps="char") _ = speech_recognizer(waveform, return_timestamps="char")
self.assertEqual(str(v.exception), "We cannot return_timestamps yet on non-ctc models !")
@require_torch @require_torch
def test_small_model_pt_seq2seq(self): def test_small_model_pt_seq2seq(self):