Up toleratce to deflake public MacOS tests

PiperOrigin-RevId: 457117919
google · Jun 25, 2022 · 6b1a929 · 6b1a929
1 parent f540981
commit 6b1a929
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/experimental/empirical_tf_test.py b/tests/experimental/empirical_tf_test.py
@@ -244,9 +244,9 @@ def _compare_ntks(
       rtol_jax = 0.15  # TODO(romann): revisit poor TPU agreement.
     else:
       atol = 1e-5
-      rtol = 1e-5
+      rtol = 5e-5
       atol_jax = 0.
-      rtol_jax = 1e-5
+      rtol_jax = 5e-5
 
     for i1, ntk1 in ntks:
       for i2, ntk2 in ntks[i1 + 1:]: