Allow n=0 in Binomial logp and logcdf

pymc-devs · Mar 15, 2022 · 09bddde · 09bddde
1 parent 08b6b01
commit 09bddde
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pymc/distributions/discrete.py b/pymc/distributions/discrete.py
@@ -146,7 +146,7 @@ def logp(value, n, p):
             binomln(n, value) + logpow(p, value) + logpow(1 - p, n - value),
         )
 
-        return check_parameters(res, 0 < n, 0 <= p, p <= 1, msg="n > 0, 0 <= p <= 1")
+        return check_parameters(res, 0 <= n, 0 <= p, p <= 1, msg="n >= 0, 0 <= p <= 1")
 
     def logcdf(value, n, p):
         """
@@ -177,10 +177,10 @@ def logcdf(value, n, p):
 
         return check_parameters(
             res,
-            0 < n,
+            0 <= n,
             0 <= p,
             p <= 1,
-            msg="n > 0, 0 <= p <= 1",
+            msg="n >= 0, 0 <= p <= 1",
         )