Skip to content

Commit

Permalink
Relax test tolerance in pinv test to fix a CI failure on Windows CPU.
Browse files Browse the repository at this point in the history
  • Loading branch information
hawkinsp authored and Google-ML-Automation committed Sep 11, 2024
1 parent e869a9d commit 2bd1fde
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/linalg_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -1152,7 +1152,7 @@ def np_fn(a):
a = (a + T(a.conj())) / 2
return np.linalg.pinv(a, hermitian=hermitian)
self._CheckAgainstNumpy(np_fn, jnp_fn, args_maker, tol=1e-4)
self._CompileAndCheck(jnp_fn, args_maker)
self._CompileAndCheck(jnp_fn, args_maker, atol=1e-5)

# TODO(phawkins): 6e-2 seems like a very loose tolerance.
jtu.check_grads(jnp_fn, args_maker(), 1, rtol=6e-2, atol=1e-3)
Expand Down

0 comments on commit 2bd1fde

Please sign in to comment.