use take instead of select

run_speech_recognition_seq2seq_streaming.py

@@ -780,7 +780,7 @@ def main():
 
     # Run a test prediction to check outputs
     predictions = trainer.predict(
-        test_dataset=vectorized_datasets["eval"].shuffle(seed=training_args.seed).select(range(5)),
+        test_dataset=vectorized_datasets["eval"].shuffle(seed=training_args.seed).take(5),
         metric_key_prefix="test",
         max_length=training_args.generation_max_length,
         num_beams=training_args.generation_num_beams,