Skip to content

Commit eb75516

Browse files
authored
Fix Fatal Python error: Bus error in ZeroShotAudioClassificationPipelineTests (huggingface#30283)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
1 parent 05dab4e commit eb75516

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/pipelines/test_pipelines_zero_shot_audio_classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def test_small_model_pt(self):
3232
audio_classifier = pipeline(
3333
task="zero-shot-audio-classification", model="hf-internal-testing/tiny-clap-htsat-unfused"
3434
)
35-
dataset = load_dataset("ashraq/esc50")
35+
dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example")
3636
audio = dataset["train"]["audio"][-1]["array"]
3737
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])
3838
self.assertEqual(
@@ -52,7 +52,7 @@ def test_large_model_pt(self):
5252
model="laion/clap-htsat-unfused",
5353
)
5454
# This is an audio of a dog
55-
dataset = load_dataset("ashraq/esc50")
55+
dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example")
5656
audio = dataset["train"]["audio"][-1]["array"]
5757
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])
5858

0 commit comments

Comments
 (0)