Use the same defaul precision as torch.gradcheck

lab-cosmo · Jul 3, 2024 · 2783dd8 · 2783dd8
1 parent 8dd7adf
commit 2783dd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/metatensor-operations/tests/_gradcheck.py b/python/metatensor-operations/tests/_gradcheck.py
@@ -11,8 +11,8 @@ def check_finite_differences(
  *,
  parameter: str,
  displacement: float = 1e-6,
- rtol: float = 1e-5,
- atol: float = 1e-15,
+ rtol: float = 1e-3,
+ atol: float = 1e-9,
 ) -> None:
  """
  Check that analytical gradients with respect to ``parameter`` in the