hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Merge remote-tracking branch 'origin/main' into user/aliberts/2025_02_10_dataset_v2.1

Browse Source
This commit is contained in:
Simon Alibert
2025-02-15 15:56:24 +01:00
parent 8426c64f42 c4c2ce04e7
commit aed3eb4a94
27 changed files with 134 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_datasets.py
View File

@@ -519,9 +519,9 @@ def test_backward_compatibility(repo_id):
assert new_keys == old_keys, f"{new_keys=} and {old_keys=} are not the same"
for key in new_frame:
assert torch.isclose(
new_frame[key], old_frame[key]
).all(), f"{key=} for index={i} does not contain the same value"
assert torch.isclose(new_frame[key], old_frame[key]).all(), (
f"{key=} for index={i} does not contain the same value"
)
# test2 first frames of first episode
i = dataset.episode_data_index["from"][0].item()