Skip to content

Commit 9f06a34

Browse files
committed
Fix docstring for semirelaxed use case
1 parent 9f0e69c commit 9f06a34

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ot/gromov/_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -405,9 +405,9 @@ def init_matrix_semirelaxed(C1, C2, p, loss_fun='square_loss', nx=None):
405405
Metric cost matrix in the source space
406406
C2 : array-like, shape (nt, nt)
407407
Metric cost matrix in the target space
408-
T : array-like, shape (ns, nt)
409-
Coupling between source and target spaces
410408
p : array-like, shape (ns,)
409+
loss_fun : str, optional
410+
Name of loss function to use: either 'square_loss' or 'kl_loss' (default='square_loss')
411411
nx : backend, optional
412412
If let to its default value None, a backend test will be conducted.
413413

0 commit comments

Comments
 (0)