From a66031f1f79f405355400536deaa4a5a701d3039 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Thu, 2 May 2024 14:36:54 +0200 Subject: [PATCH] Remove default params torch.isclose --- tests/test_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index e4be423f..1783b427 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -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