Small bug fix on test.py

verlab · Apr 22, 2023 · d1451c6 · d1451c6
1 parent 3e6ce97
commit d1451c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rl_fast_forward/test.py b/rl_fast_forward/test.py
@@ -204,7 +204,7 @@ def test(args, test_envs, annotations):
 
     elif args.dataset:
 
-        experiments = np.array(Experiment2VideoMapping.get_dataset_experiments(args.dataset, cvpr2020_subset=args.cvpr2020_subset))[:10]
+        experiments = np.array(Experiment2VideoMapping.get_dataset_experiments(args.dataset, cvpr2020_subset=args.cvpr2020_subset))
         random.shuffle(experiments)
 
         annotations = json.load(open(config.annotations_filenames[args.dataset]))