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

Fix tolerance on bernoulli test

Browse Source
This commit is contained in:
Jake VanderPlas 2023-08-23 16:59:27 -07:00
parent f62b9eb3d1
commit 75d12a2e21
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/lax_scipy_special_functions_test.py
View File

@ -194,7 +194,7 @@ class LaxScipySpcialFunctionsTest(jtu.JaxTestCase):
lax_op = functools.partial(lsp_special.bernoulli, n)
args_maker = lambda: []
self._CheckAgainstNumpy(scipy_op, lax_op, args_maker, atol=0, rtol=1E-5)
self._CompileAndCheck(lax_op, args_maker)
self._CompileAndCheck(lax_op, args_maker, atol=0, rtol=1E-5)
if __name__ == "__main__":