From 6928cd415d1c16139ce9a6a1c373b9d5222c730a Mon Sep 17 00:00:00 2001 From: Balakumar Sundaralingam Date: Fri, 23 Feb 2024 09:03:23 -0800 Subject: [PATCH] minor test fix --- tests/goal_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/goal_test.py b/tests/goal_test.py index 740b9c9..b95ac1b 100644 --- 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