From 2bd1fdead81581db08ee84a0d1f82c407ccd6b11 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Wed, 11 Sep 2024 08:49:23 -0700
Subject: [PATCH] Relax test tolerance in pinv test to fix a CI failure on
 Windows CPU.

https://github.com/google/jax/actions/runs/10812364182/job/29993831201

PiperOrigin-RevId: 673409820
---
 tests/linalg_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/linalg_test.py b/tests/linalg_test.py
index 901bfca997dc..4dcdeb19e1ef 100644
--- a/tests/linalg_test.py
+++ b/tests/linalg_test.py
@@ -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)