Bernoulli docstring update as in pymc-devs#5459

mariyayb · Jul 9, 2022 · 17edf24 · 17edf24
1 parent 966aabf
commit 17edf24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymc/distributions/discrete.py b/pymc/distributions/discrete.py
@@ -366,9 +366,9 @@ class Bernoulli(Discrete):
 
     Parameters
     ----------
-    p: float
+    p : tensor_like or float
         Probability of success (0 < p < 1).
-    logit_p: float
+    logit_p : tensor_like or float
         Alternative log odds for the probability of success.
     """
     rv_op = bernoulli