Skip to content

Commit 7371d2a

Browse files
committed
Update atol for platform precision
1 parent bc111ca commit 7371d2a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pymc/tests/distributions/util.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,7 @@ def check_logp(
279279
rtol = select_by_precision()
280280

281281
if atol is None:
282-
atol = 1e-10
282+
atol = select_by_precision(float64=1e-10, float32=1e-8)
283283

284284
if extra_args is None:
285285
extra_args = {}
@@ -463,7 +463,7 @@ def check_logcdf(
463463
rtol = select_by_precision()
464464

465465
if atol is None:
466-
atol = 1e-10
466+
atol = select_by_precision(float64=1e-10, float32=1e-8)
467467

468468
for pt in product(domains, n_samples=n_samples):
469469
params = dict(pt)
@@ -563,7 +563,7 @@ def check_selfconsistency_discrete_logcdf(
563563
rtol = select_by_precision(float64=1e-8, float32=1e-5)
564564

565565
if atol is None:
566-
atol = 0.0
566+
atol = select_by_precision(float64=1e-10, float32=1e-8)
567567

568568
model, param_vars = build_model(distribution, domain, paramdomains)
569569
rv = model["value"]

0 commit comments

Comments
 (0)