diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 6d0055a2..2e0686e4 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -128,6 +128,7 @@ def test_multilerobotdataset_frames(): expected_dataset_indices = [] for i, sub_dataset in enumerate(sub_datasets): expected_dataset_indices.extend([i] * len(sub_dataset)) + for expected_dataset_index, sub_dataset_item, dataset_item in zip( expected_dataset_indices, chain(*sub_datasets), dataset, strict=True ):