fixed the docs about the implicit_diff argument (#270)

ott-jax · Feb 14, 2023 · 3ab8c28 · 3ab8c28
1 parent 2cef1c6
commit 3ab8c28
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ott/solvers/linear/sinkhorn.py b/src/ott/solvers/linear/sinkhorn.py
@@ -574,8 +574,8 @@ class Sinkhorn:
     The optimal solutions ``f`` and ``g`` and the optimal objective
     (``reg_ot_cost``) outputted by the Sinkhorn algorithm can be differentiated
     w.r.t. relevant inputs ``geom``, ``a`` and ``b`` using, by default, implicit
-    differentiation of the optimality conditions (``implicit_differentiation``
-    set to ``True``). This choice has two consequences.
+    differentiation of the optimality conditions (``implicit_diff``
+    not equal to ``None``). This choice has two consequences.
 
     - The termination criterion used to stop Sinkhorn (cancellation of
       gradient of objective w.r.t. ``f_u`` and ``g_v``) is used to differentiate