From 70f186f61ebff4ad87fd7cc0fc1e0e0660b485ea Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 22 Oct 2021 11:01:26 +0200 Subject: [PATCH] up (#14116) --- tests/test_modeling_hubert.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_modeling_hubert.py b/tests/test_modeling_hubert.py index 5d3d27721e..ad8a1dd206 100644 --- 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]