wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor change to improve test clarity

This commit is contained in:
Radek Osmulski 2024-05-28 23:16:15 +10:00
parent 621f69d98f
commit 5d7a4e316b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_sampler.py
View File

@ -85,5 +85,6 @@ def test_shuffle():
assert len(sampler) == 6 assert len(sampler) == 6
assert list(sampler) == [0, 1, 2, 3, 4, 5] assert list(sampler) == [0, 1, 2, 3, 4, 5]
sampler = EpisodeAwareSampler(episode_data_index, shuffle=True) sampler = EpisodeAwareSampler(episode_data_index, shuffle=True)
assert sampler.indices == [0, 1, 2, 3, 4, 5]
assert len(sampler) == 6 assert len(sampler) == 6
assert set(sampler) == {0, 1, 2, 3, 4, 5} assert set(sampler) == {0, 1, 2, 3, 4, 5}