Check only whether problem can be LR (#248)

* Check only whether problem can be LR * Fix indentation in `docs`
ott-jax · Feb 9, 2023 · de2c1d7 · de2c1d7
1 parent a836f4b
commit de2c1d7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/ott/solvers/quadratic/gromov_wasserstein.py b/src/ott/solvers/quadratic/gromov_wasserstein.py
@@ -215,8 +215,7 @@ def __call__(
       key = jax.random.PRNGKey(0)
     key1, key2 = jax.random.split(key, 2)
 
-    # consider converting problem first if using low-rank solver
-    if self.is_low_rank and prob._is_low_rank_convertible:
+    if prob._is_low_rank_convertible:
       prob = prob.to_low_rank()
 
     if init is None: