set a lower pval for vmap-of-rbg based uniform statistical test

PiperOrigin-RevId: 614434781
jax-ml · Mar 10, 2024 · 477a5aa · 477a5aa
1 parent 63ceb5f
commit 477a5aa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/random_lax_test.py b/tests/random_lax_test.py
@@ -1459,7 +1459,8 @@ def rand(key):
 
     for samples in [uncompiled_samples, compiled_samples]:
       self._CheckCollisions(samples, jnp.finfo(dtype).nmant)
-      self._CheckKolmogorovSmirnovCDF(samples, scipy.stats.uniform().cdf)
+      self._CheckKolmogorovSmirnovCDF(samples, scipy.stats.uniform().cdf,
+                                      pval=0.005)
 
   def test_cannot_add(self):
     key = self.make_key(73)