up (#14116)

70f186f6 · Patrick von Platen · GitHub · ca2ef7df · 70f186f6
Unverified Commit 70f186f6 authored Oct 22, 2021 by Patrick von Platen Committed by GitHub Oct 22, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

tests/test_modeling_hubert.py tests/test_modeling_hubert.py +3 -1

No files found.
--- a/tests/test_modeling_hubert.py
+++ b/tests/test_modeling_hubert.py
@@ -615,7 +615,9 @@ class HubertModelIntegrationTest(unittest.TestCase):

        ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
        # automatic decoding with librispeech
-        speech_samples = ds.sort("id").select(range(num_samples))[:num_samples]["audio"]
+        speech_samples = ds.sort("id").filter(
+            lambda x: x["id"] in [f"1272-141231-000{i}" for i in range(num_samples)]
+        )[:num_samples]["audio"]

        return [x["array"] for x in speech_samples]