adapt test

theislab · Oct 9, 2023 · 4aab671 · 4aab671
1 parent b233601
commit 4aab671
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/backends/ott/test_backend.py b/tests/backends/ott/test_backend.py
@@ -129,12 +129,11 @@ def test_epsilon(self, x_cost: jnp.ndarray, y_cost: jnp.ndarray, eps: Optional[f
         assert isinstance(solver.y, Geometry)
         np.testing.assert_allclose(gt.matrix, pred.transport_matrix, rtol=RTOL, atol=ATOL)
 
-    @pytest.mark.skip(reason="TODO")
     @pytest.mark.parametrize("rank", [-1, 7])
     def test_solver_rank(self, x: Geom_t, y: Geom_t, rank: int) -> None:
         thresh, eps = 1e-2, 1e-2
         if rank > -1:
-            gt = LRGromovWasserstein(epsilon=eps, rank=rank, threshold=thresh)(
+            gt = LRGromovWasserstein(epsilon=eps, rank=rank, threshold=thresh, initializer="rank2")(
                 QuadraticProblem(PointCloud(x, epsilon=eps), PointCloud(y, epsilon=eps))
             )