Skip to content

Commit 686811c

Browse files
authored
Fix docstrings reference for the betainc_grad function (#412)
1 parent b968766 commit 686811c

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

pytensor/scalar/math.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1357,13 +1357,13 @@ def c_code(self, *args, **kwargs):
13571357

13581358

13591359
def betainc_grad(p, q, x, wrtp: bool):
1360-
"""Gradient of the regularized lower gamma function (P) wrt to the first
1361-
argument (k, a.k.a. alpha).
1362-
1363-
Adapted from STAN `grad_reg_lower_inc_gamma.hpp`
1360+
"""
1361+
Gradient of the regularized incomplete beta function wrt to the first
1362+
argument `p` (aka alpha) or the second argument `q` (aka beta),
1363+
depending on whether `wrtp` is true.
13641364
1365-
Reference: Gautschi, W. (1979). A computational procedure for incomplete gamma functions.
1366-
ACM Transactions on Mathematical Software (TOMS), 5(4), 466-481.
1365+
Reference: Boik, R. J., & Robison-Cox, J. F. (1998). Derivatives of the incomplete beta function.
1366+
Journal of Statistical Software, 3(1), 1-20.
13671367
"""
13681368

13691369
def _betainc_der(p, q, x, wrtp, skip_loop):

0 commit comments

Comments
 (0)