From e5fa8bbc644475c7ca954892513aa72e474c3683 Mon Sep 17 00:00:00 2001 From: Charles MOUSSA Date: Thu, 8 Aug 2024 10:49:05 +0200 Subject: [PATCH] using rtol for hamevo gpsr --- 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 0e8307ec..1e2fbd97 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 ) - atol = 1.0e-01 + rtol = 1.0e-02 # first order checks for i in range(len(grad_ad)): - assert torch.allclose(grad_ad[i], grad_gpsr[i], atol=atol) + assert torch.allclose(grad_ad[i], grad_gpsr[i], rtol=rtol) @pytest.mark.parametrize(