Unverified Commit eb75516e authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `Fatal Python error: Bus error` in `ZeroShotAudioClassificationPipelineTests` (#30283)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 05dab4e5
...@@ -32,7 +32,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase): ...@@ -32,7 +32,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase):
audio_classifier = pipeline( audio_classifier = pipeline(
task="zero-shot-audio-classification", model="hf-internal-testing/tiny-clap-htsat-unfused" 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"] audio = dataset["train"]["audio"][-1]["array"]
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])
self.assertEqual( self.assertEqual(
...@@ -52,7 +52,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase): ...@@ -52,7 +52,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase):
model="laion/clap-htsat-unfused", model="laion/clap-htsat-unfused",
) )
# This is an audio of a dog # 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"] audio = dataset["train"]["audio"][-1]["array"]
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment