[ci skip] Make to_LRCGeometry in LR geom no-op

ott-jax · Jul 7, 2022 · fe7df9a · fe7df9a
1 parent 139a53b
commit fe7df9a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ott/geometry/low_rank.py b/ott/geometry/low_rank.py
@@ -221,6 +221,12 @@ def finalize(carry):
     max_value = jnp.max(jnp.concatenate((out, last_slice.reshape(-1))))
     return max_value + self._bias
 
+  def to_LRCGeometry(
+      self, rank: int, tol: float = 1e-2, seed: int = 0
+  ) -> 'LRCGeometry':
+    """Return self."""
+    return self
+
   def tree_flatten(self):
     return (self._cost_1, self._cost_2, self._kwargs), {
         'bias': self._bias,