Skip to content

Commit

Permalink
reput atol 0.1
Browse files Browse the repository at this point in the history
  • Loading branch information
Charles MOUSSA committed Aug 8, 2024
1 parent 474d7c6 commit d957514
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_gpsr.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit d957514

Please sign in to comment.