mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Relax test tolerances for testLogisticPpf.

Browse Source 
Fixes a test failure in CI.

PiperOrigin-RevId: 520649225
This commit is contained in:
Peter Hawkins 2023-03-30 08:41:10 -07:00 committed by jax authors
parent dedfc8df75
commit 67a28ce30f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/scipy_stats_test.py
View File

@ -603,7 +603,7 @@ class LaxBackedScipyStatsTests(jtu.JaxTestCase):
with jtu.strict_promotion_if_dtypes_match(dtypes):
self._CheckAgainstNumpy(scipy_fun, lax_fun, args_maker, check_dtypes=False,
tol=1e-4)
atol=1e-3, rtol=1e-3)
self._CompileAndCheck(lax_fun, args_maker, rtol=3e-4)
@genNamedParametersNArgs(3)