Skip to content

Commit b2b8f6c

Browse files
authored
Update docstring to include 'a' parameter
Added parameter description for 'a' in the docstring.
1 parent 5a7d086 commit b2b8f6c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ot/stochastic.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,8 @@ def averaged_sgd_entropic_transport(
178178
179179
Parameters
180180
----------
181+
a : ndarray, shape (ns,)
182+
source measure
181183
b : ndarray, shape (nt,)
182184
target measure
183185
M : ndarray, shape (ns, nt)

0 commit comments

Comments
 (0)