Skip to content

Commit c61ef17

Browse files
committed
fix random.loggamma compatibility
1 parent 51e6df5 commit c61ef17

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

brainpy/math/random.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1343,8 +1343,22 @@ def orthogonal(n: int, size=None):
13431343
return DEFAULT.orthogonal(n, size)
13441344

13451345

1346-
@wraps(jr.loggamma)
13471346
def loggamma(a, size=None):
1347+
"""Sample log-gamma random values.
1348+
1349+
Parameters
1350+
----------
1351+
a: float, array_like
1352+
A float or array of floats broadcast-compatible with shape representing the parameter of the distribution.
1353+
size: optional, int, tuple of int
1354+
A tuple of nonnegative integers specifying the result shape.
1355+
Must be broadcast-compatible with `a`. The default (None) produces a result shape equal to `a.shape`.
1356+
1357+
Returns
1358+
-------
1359+
out: array_like
1360+
The sampled results.
1361+
"""
13481362
return DEFAULT.loggamma(a, size)
13491363

13501364

0 commit comments

Comments
 (0)