From d658aafac4d5ff7b7dcf033c7ee1941d23ec7afa Mon Sep 17 00:00:00 2001 From: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> Date: Fri, 12 Jul 2024 08:26:00 +0200 Subject: [PATCH] Mark test in test_upstream_hub with require_librosa --- tests/test_upstream_hub.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_upstream_hub.py b/tests/test_upstream_hub.py index daa0841ac98..1e15584f41e 100644 --- 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")