diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 2e0686e4..dac18c14 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -123,6 +123,7 @@ def test_multilerobotdataset_frames(): assert len(dataset) == sum(len(d) for d in sub_datasets) assert dataset.num_samples == sum(d.num_samples for d in sub_datasets) assert dataset.num_episodes == sum(d.num_episodes for d in sub_datasets) + # Run through all items of the LeRobotDatasets in parallel with the items of the MultiLerobotDataset and # check they match. expected_dataset_indices = []