minor test fix

2024-02-23 09:03:23 -08:00
parent bafdf80c05
commit 6928cd415d
1 changed files with 2 additions and 2 deletions
--- a/tests/goal_test.py
+++ b/tests/goal_test.py
@@ -82,7 +82,7 @@ def test_repeat_seeds():
        True,
    )

-    assert torch.sum(r[0]).item() == 0.0
+    assert torch.sum(r[0]).item() <= 1e-5


 def test_horizon_repeat_seeds():
@@ -153,4 +153,4 @@ def test_horizon_repeat_seeds():
        False,
        True,
    )
-    assert torch.sum(r[0]).item() < 1e-6
+    assert torch.sum(r[0]).item() < 1e-5