Skip to content

Commit 7fe5164

Browse files
authored
correct docstrings in Binomial Class (#5952)
* correct docstrings to add tensor_like * add close-figs context Co-authored-by: SangamSwadiK <[email protected]>
1 parent dfd018e commit 7fe5164

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pymc/distributions/discrete.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ class Binomial(Discrete):
8282
.. math:: f(x \mid n, p) = \binom{n}{x} p^x (1-p)^{n-x}
8383
8484
.. plot::
85+
:context: close-figs
8586
8687
import matplotlib.pyplot as plt
8788
import numpy as np
@@ -107,11 +108,11 @@ class Binomial(Discrete):
107108
108109
Parameters
109110
----------
110-
n : int
111+
n : tensor_like of int
111112
Number of Bernoulli trials (n >= 0).
112-
p : float
113+
p : tensor_like of float
113114
Probability of success in each trial (0 < p < 1).
114-
logit_p : float
115+
logit_p : tensor_like of float
115116
Alternative log odds for the probability of success.
116117
"""
117118
rv_op = binomial

0 commit comments

Comments
 (0)