remove atol from test

alexhernandezgarcia · Aug 26, 2024 · 42b1913 · 42b1913
1 parent 0ef0d1c
commit 42b1913
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/gflownet/utils/test_batch.py b/tests/gflownet/utils/test_batch.py
@@ -1641,7 +1641,7 @@ def test__get_logrewards_multiple_env_returns_expected_non_zero_non_terminating(
         )
     ), (logrewards, logrewards_batch)
     assert ~torch.any(
-        torch.isclose(logrewards_batch, torch.zeros_like(logrewards_batch), atol=1e-10)
+        torch.isclose(logrewards_batch, torch.zeros_like(logrewards_batch))
     ), logrewards_batch