Relax tests

PiperOrigin-RevId: 529841759
google · May 8, 2023 · 928b0bc · 928b0bc
1 parent ed115c7
commit 928b0bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/stax/branching_test.py b/tests/stax/branching_test.py
@@ -199,7 +199,7 @@ def test_fan_in_conv(
     else:
       width = 1024
       n_samples = 512
-      tol = 0.01
+      tol = 0.015
 
     conv = stax.Conv(out_chan=width,
                      filter_shape=(3, 3),