Skip to content

Commit ec70d86

Browse files
kataevDenis Kataev
and
Denis Kataev
authored
Fix docs formatting in shape_utils (#7025)
* Small docs update shape_utils.py Fix docs markup * Update to_tuple docs shape_utils.py * Types --------- Co-authored-by: Denis Kataev <[email protected]>
1 parent 373274a commit ec70d86

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

pymc/distributions/shape_utils.py

+7-5
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,9 @@ def to_tuple(shape):
5959
6060
Returns
6161
-------
62-
If `shape` is None, returns an empty tuple. If it's an int, (shape,) is
63-
returned. If it is array-like, tuple(shape) is returned.
62+
shape : tuple
63+
If `shape` is None, returns an empty tuple. If it's an int, (shape,) is
64+
returned. If it is array-like, tuple(shape) is returned.
6465
"""
6566
if shape is None:
6667
return tuple()
@@ -321,9 +322,10 @@ def change_dist_size(
321322
322323
Returns
323324
-------
324-
A new distribution variable that is equivalent to the original distribution with
325-
the new size. The new distribution will not reuse the old RandomState/Generator
326-
input, so it will be independent from the original distribution.
325+
dist : TensorVariable
326+
A new distribution variable that is equivalent to the original distribution with
327+
the new size. The new distribution will not reuse the old RandomState/Generator
328+
input, so it will be independent from the original distribution.
327329
328330
Examples
329331
--------

0 commit comments

Comments
 (0)