Changed at.eq to at.isclose in logp as Ricardo suggested.

pymc-devs · Jan 19, 2022 · d563930 · d563930
1 parent 1bdc442
commit d563930
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc/distributions/multivariate.py b/pymc/distributions/multivariate.py
@@ -583,7 +583,7 @@ def logp(value, n, p):
         return check_parameters(
             res,
             p <= 1,
-            at.eq(at.sum(p, axis=-1), 1),
+            at.isclose(at.sum(p, axis=-1), 1),
             at.ge(n, 0),
             msg="p <= 1, sum(p) = 1, n >= 0",
         )