From d9575146486a7568dff8aa095ca5a752503189fd Mon Sep 17 00:00:00 2001 From: Charles MOUSSA Date: Thu, 8 Aug 2024 11:06:31 +0200 Subject: [PATCH] reput atol 0.1 --- tests/test_gpsr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_gpsr.py b/tests/test_gpsr.py index ae2cba69..0e8307ec 100644 --- a/tests/test_gpsr.py +++ b/tests/test_gpsr.py @@ -138,12 +138,12 @@ def test_expectation_gpsr_hamevo( exp_gpsr, tuple(values.values()), torch.ones_like(exp_gpsr), create_graph=True ) - rtol = 1.0e-01 + atol = 1.0e-01 # first order checks for i in range(len(grad_ad)): - assert torch.allclose(grad_ad[i], grad_gpsr[i], rtol=rtol) + assert torch.allclose(grad_ad[i], grad_gpsr[i], atol=atol) @pytest.mark.parametrize(