Enable ZeroShotAudioClassificationPipelineTests::test_small_model_pt (#24882)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-07-18 15:08:53 +02:00
committed by GitHub
parent 9c875839c0
commit 57da42ad05

View File

@@ -27,8 +27,6 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase):
# and only CLAP would be there for now. # and only CLAP would be there for now.
# model_mapping = {CLAPConfig: CLAPModel} # model_mapping = {CLAPConfig: CLAPModel}
# TODO: fix me (ydshieh)
@unittest.skip("currently failing (probably due to `datasets` issue)")
@require_torch @require_torch
def test_small_model_pt(self): def test_small_model_pt(self):
audio_classifier = pipeline( audio_classifier = pipeline(