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

View File

@ -269,7 +269,7 @@ def test_backward_compatibility(repo_id):
for key in new_frame:
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"
# test2 first frames of first episode