minor test fix

This commit is contained in:
Balakumar Sundaralingam
2024-02-23 09:03:23 -08:00
parent bafdf80c05
commit 6928cd415d

View File

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