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

Remove default params torch.isclose

This commit is contained in:
Simon Alibert 2024-05-02 14:36:54 +02:00
parent 14648f81d0
commit a66031f1f7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_datasets.py
View File

@ -269,7 +269,7 @@ def test_backward_compatibility(repo_id):
for key in new_frame: for key in new_frame:
assert torch.isclose( assert torch.isclose(
new_frame[key], old_frame[key], rtol=1e-05, atol=1e-08 new_frame[key], old_frame[key]
).all(), f"{key=} for index={i} does not contain the same value" ).all(), f"{key=} for index={i} does not contain the same value"
# test2 first frames of first episode # test2 first frames of first episode