Merge pull request jax-ml#3395 from google/tweak-tolerance

tweak t logpdf tolerance for float64
BuddenD · Jun 10, 2020 · e6a08e2 · e6a08e2
2 parents b7e945f + 650fb49
commit e6a08e2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/scipy_stats_test.py b/tests/scipy_stats_test.py
@@ -373,7 +373,8 @@ def args_maker():
 
     self._CheckAgainstNumpy(scipy_fun, lax_fun, args_maker, check_dtypes=False,
                             tol=1e-3)
-    self._CompileAndCheck(lax_fun, args_maker)
+    self._CompileAndCheck(lax_fun, args_maker,
+                          rtol={onp.float64: 1e-14}, atol={onp.float64: 1e-14})
 
 
   @genNamedParametersNArgs(3, jtu.rand_default)