[ci-daily] Fix pipeline tests (#21257)

* use streaming dataset

* fix whisper's test

* add rescale argument to chunk_iter
This commit is contained in:
Arthur
2023-01-23 19:32:49 +01:00
committed by GitHub
parent 275ad9d80a
commit b80b2218b5
2 changed files with 21 additions and 15 deletions

View File

@@ -201,8 +201,9 @@ class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase, metaclass=Pipel
@require_torch
@require_pyctcdecode
def test_large_model_pt_with_lm(self):
dataset = load_dataset("Narsil/asr_dummy")
filename = dataset["test"][3]["file"]
dataset = load_dataset("Narsil/asr_dummy", streaming=True)
third_item = next(iter(dataset["test"].skip(3)))
filename = third_item["file"]
speech_recognizer = pipeline(
task="automatic-speech-recognition",