From eb75516e7cf1b7673ec39cc0223ad6f6673e06ef Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 17 Apr 2024 11:47:30 +0200 Subject: [PATCH] Fix `Fatal Python error: Bus error` in `ZeroShotAudioClassificationPipelineTests` (#30283) fix Co-authored-by: ydshieh --- .../test_pipelines_zero_shot_audio_classification.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/test_pipelines_zero_shot_audio_classification.py b/tests/pipelines/test_pipelines_zero_shot_audio_classification.py index 87f91a7d27..80414f1722 100644 --- a/tests/pipelines/test_pipelines_zero_shot_audio_classification.py +++ b/tests/pipelines/test_pipelines_zero_shot_audio_classification.py @@ -32,7 +32,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase): audio_classifier = pipeline( task="zero-shot-audio-classification", model="hf-internal-testing/tiny-clap-htsat-unfused" ) - dataset = load_dataset("ashraq/esc50") + dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example") audio = dataset["train"]["audio"][-1]["array"] output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) self.assertEqual( @@ -52,7 +52,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase): model="laion/clap-htsat-unfused", ) # This is an audio of a dog - dataset = load_dataset("ashraq/esc50") + dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example") audio = dataset["train"]["audio"][-1]["array"] output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])