Skip to content

Commit 8787fc2

Browse files
authored
Fix incorrect statement in GammaPoisson log_prob docstring (#2194)
The mathematical expression is for the unlogged pmf, not the log pmf.
1 parent a1bcf5e commit 8787fc2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

numpyro/distributions/conjugate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -383,7 +383,7 @@ def sample(self, key: jax.Array, sample_shape: tuple[int, ...] = ()) -> ArrayLik
383383

384384
@validate_sample
385385
def log_prob(self, value: ArrayLike) -> ArrayLike:
386-
r"""If :math:`X \sim \mathrm{GammaPoisson}(\alpha, \lambda)`, then the log
386+
r"""If :math:`X \sim \mathrm{GammaPoisson}(\alpha, \lambda)`, then the
387387
probability mass function is:
388388
389389
.. math::

0 commit comments

Comments
 (0)