Increase tolerance of test in gumbel_test.py.

PiperOrigin-RevId: 502797940
google-deepmind · Jan 18, 2023 · c82a2ea · c82a2ea
1 parent c47c62f
commit c82a2ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distrax/_src/distributions/gumbel_test.py b/distrax/_src/distributions/gumbel_test.py
@@ -182,7 +182,7 @@ def test_with_two_distributions(self, function_string, mode_string):
             'loc': np.array(np.random.randn(3, 2), dtype=np.float32),
             'scale': 0.1 + np.array(np.random.rand(4, 1, 2), dtype=np.float32),
         },
-        assertion_fn=self.assertion_fn(rtol=5e-2))
+        assertion_fn=self.assertion_fn(rtol=6e-2))
 
   def test_jittable(self):
     super()._test_jittable(