Update tests/test_datasets.py
Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
parent
aad56631a8
commit
984810462f
|
@ -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 = []
|
||||
|
|
Loading…
Reference in New Issue