From aad56631a8488fc49160170b089a77ece6b0e770 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Thu, 30 May 2024 17:41:32 +0100 Subject: [PATCH] Update tests/test_datasets.py Co-authored-by: Remi --- tests/test_datasets.py | 1 + 1 file changed, 1 insertion(+) 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 ):