Skip to content

Commit d3f5bf3

Browse files
committed
small stuff
1 parent c098a15 commit d3f5bf3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ot/solvers.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,11 @@ def solve(M, a=None, b=None, reg=None, reg_type="KL", unbalanced=None,
5959
Unbalanced penalization weight :math:`\lambda_u`, by default None
6060
(balanced OT)
6161
unbalanced_type : str, optional
62-
Type of unbalanced penalization function :math:`U` either "KL", "L2", "TV", by default "KL"
62+
Type of unbalanced penalization function :math:`U` either "KL", "L2",
63+
"TV", by default "KL"
64+
method : str, optional
65+
Method for solving the problem when multiple algorithms are available,
66+
default None for automatic selection.
6367
n_threads : int, optional
6468
Number of OMP threads for exact OT solver, by default 1
6569
max_iter : int, optional

0 commit comments

Comments
 (0)