Skip to content

Commit 7ee082d

Browse files
committed
Fix docstring for loss_fun parameter
1 parent 7fa8438 commit 7ee082d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ot/gromov/_bregman.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -373,8 +373,8 @@ def entropic_gromov_barycenters(
373373
lambdas : list of float, optional
374374
List of the `S` spaces' weights.
375375
If let to its default value None, uniform weights are taken.
376-
loss_fun : callable, optional
377-
tensor-matrix multiplication function based on specific loss function
376+
loss_fun : string, optional (default='square_loss')
377+
Loss function used for the solver either 'square_loss' or 'kl_loss'
378378
epsilon : float, optional
379379
Regularization term >0
380380
symmetric : bool, optional.

0 commit comments

Comments
 (0)