Mark test in test_upstream_hub with require_librosa

huggingface · Jul 12, 2024 · d658aaf · d658aaf
1 parent 5f42709
commit d658aaf
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_upstream_hub.py b/tests/test_upstream_hub.py
@@ -34,8 +34,9 @@
 )
 from datasets.utils.file_utils import cached_path
 from datasets.utils.hub import hf_dataset_url
-from tests.fixtures.hub import CI_HUB_ENDPOINT, CI_HUB_USER, CI_HUB_USER_TOKEN
-from tests.utils import for_all_test_methods, require_pil, require_sndfile, xfail_if_500_502_http_error
+
+from .fixtures.hub import CI_HUB_ENDPOINT, CI_HUB_USER, CI_HUB_USER_TOKEN
+from .utils import for_all_test_methods, require_librosa, require_pil, require_sndfile, xfail_if_500_502_http_error
 
 
 pytestmark = pytest.mark.integration
@@ -383,6 +384,7 @@ def test_push_dataset_to_hub_custom_features(self, temporary_repo):
             assert ds.features == hub_ds.features
             assert ds[:] == hub_ds[:]
 
+    @require_librosa
     @require_sndfile
     def test_push_dataset_to_hub_custom_features_audio(self, temporary_repo):
         audio_path = os.path.join(os.path.dirname(__file__), "features", "data", "test_audio_44100.wav")