Remove datasets requirement (#14795)

2021-12-16 14:34:14 -05:00
parent bef1e3e4a0
commit d194d639ab
19 changed files with 11 additions and 65 deletions
--- a/tests/test_pipelines_audio_classification.py
+++ b/tests/test_pipelines_audio_classification.py
@@ -21,7 +21,6 @@ from transformers.pipelines import AudioClassificationPipeline, pipeline
 from transformers.testing_utils import (
    is_pipeline_test,
    nested_simplify,
-    require_datasets,
    require_tf,
    require_torch,
    require_torchaudio,
@@ -65,7 +64,6 @@ class AudioClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest

        self.run_torchaudio(audio_classifier)

-    @require_datasets
    @require_torchaudio
    def run_torchaudio(self, audio_classifier):
        import datasets
@@ -101,7 +99,6 @@ class AudioClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest
        )

    @require_torch
-    @require_datasets
    @slow
    def test_large_model_pt(self):
        import datasets