admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 02:59:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests) add features argument to load_nested_dataset

Browse Source
This commit is contained in:
Michel Aractingi
2025-07-05 10:16:29 +02:00
parent 3dbc3e60fb
commit 83bf24cc9a
4 changed files with 11 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/utils/test_replay_buffer.py
+1 -1
View File
@@ -384,7 +384,7 @@ def test_to_lerobot_dataset(tmp_path):
elif feature == "next.done":
assert torch.equal(value, buffer.dones[i])
elif feature == "observation.image":
# Tenssor -> numpy is not precise, so we have some diff there
# Tensor -> numpy is not precise, so we have some diff there
# TODO: Check and fix it
torch.testing.assert_close(value, buffer.states["observation.image"][i], rtol=0.3, atol=0.003)
elif feature == "observation.state":