From 984810462f829086f4cc4bcd03a08a955b118025 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Thu, 30 May 2024 17:41:49 +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 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 = []