Skip to content

Commit 631d3d4

Browse files
cschefflerbchen93
authored andcommitted
Fix docstring of the ZeroInflatedPoisson distribution.
The parameterization uses the variable `mu` but the docstring uses `mu` in some cases and `theta` in other cases. The fix replaces all `theta` with `mu` to match the argument name for this distribution.
1 parent b6caad8 commit 631d3d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc/distributions/discrete.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1465,7 +1465,7 @@ class ZeroInflatedPoisson:
14651465
14661466
f(x \mid \psi, \mu) = \left\{ \begin{array}{l}
14671467
(1-\psi) + \psi e^{-\mu}, \text{if } x = 0 \\
1468-
\psi \frac{e^{-\theta}\theta^x}{x!}, \text{if } x=1,2,3,\ldots
1468+
\psi \frac{e^{-\mu}\mu^x}{x!}, \text{if } x=1,2,3,\ldots
14691469
\end{array} \right.
14701470
14711471
.. plot::

0 commit comments

Comments
 (0)