Skip to content

Commit b647737

Browse files
saurbhcaloctavodia
authored andcommitted
update docstrings in ZeroInflatedPoisson, DiracDelta and OrderedLogistic classes
1 parent a9baa42 commit b647737

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pymc/distributions/discrete.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1371,7 +1371,7 @@ class DiracDelta(Discrete):
13711371
13721372
Parameters
13731373
----------
1374-
c: float or int
1374+
c : tensor_like of float or int
13751375
Dirac Delta parameter. The dtype of `c` determines the dtype of the distribution.
13761376
This can affect which sampler is assigned to DiracDelta variables, or variables
13771377
that use DiracDelta, such as Mixtures.
@@ -1497,9 +1497,9 @@ class ZeroInflatedPoisson:
14971497
14981498
Parameters
14991499
----------
1500-
psi: float
1500+
psi : tensor_like of float
15011501
Expected proportion of Poisson variates (0 < psi < 1)
1502-
mu: float
1502+
mu : tensor_like of float
15031503
Expected number of occurrences during the given interval
15041504
(mu >= 0).
15051505
"""
@@ -1732,9 +1732,9 @@ class OrderedLogistic:
17321732
17331733
Parameters
17341734
----------
1735-
eta: float
1735+
eta : tensor_like of float
17361736
The predictor.
1737-
cutpoints: array
1737+
cutpoints : tensor_like of array
17381738
The length K - 1 array of cutpoints which break :math:`\eta` into
17391739
ranges. Do not explicitly set the first and last elements of
17401740
:math:`c` to negative and positive infinity.

0 commit comments

Comments
 (0)