Fix docs phrasing

ott-jax · Nov 25, 2024 · e3260cc · e3260cc
1 parent 93e8ad0
commit e3260cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ott/initializers/linear/initializers_lr.py b/src/ott/initializers/linear/initializers_lr.py
@@ -95,7 +95,7 @@ def init_r(
     """Initialize the low-rank factor :math:`R`.
 
     Args:
-      ot_prob: Linear OT problem.
+      ot_prob: OT problem.
       rng: Random key for seeding.
       init_g: Initial value for :math:`g` factor.
       kwargs: Additional keyword arguments.
@@ -131,7 +131,7 @@ def __call__(
     """Initialize the factors :math:`Q`, :math:`R` and :math:`g`.
 
     Args:
-      ot_prob: Linear OT problem.
+      ot_prob: OT problem.
       rng: Random key for seeding.
       kwargs: Additional keyword arguments for :meth:`init_q`, :meth:`init_r`
         and :meth:`init_g`.