Skip to content

Commit 8685005

Browse files
committed
revert alpha lim
1 parent d2e72b5 commit 8685005

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc_extras/distributions/discrete.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def rng_fn(cls, rng, r, alpha, size):
423423

424424
output = np.zeros(shape=size + (1,)) # noqa:RUF005
425425

426-
lam = rng.gamma(shape=r, scale=1 / alpha, size=size)
426+
lam = rng.gamma(shape=r, scale=1/alpha, size=size)
427427

428428
def sim_data(lam):
429429
# TODO: To support time-varying covariates, covariate vector may need to be added
@@ -517,7 +517,7 @@ def logcdf(value, r, alpha):
517517
return check_parameters(
518518
logcdf,
519519
r > 0,
520-
alpha > 0.6181, # alpha must be greater than 0.6181 for convergence
520+
alpha > 0, # alpha must be greater than 0.6181 for convergence
521521
msg="r > 0, alpha > 0",
522522
)
523523

0 commit comments

Comments
 (0)