see #678

theislab · Apr 30, 2024 · 82737c9 · 82737c9
1 parent db94c44
commit 82737c9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/problems/space/test_alignment_problem.py b/tests/problems/space/test_alignment_problem.py
@@ -86,7 +86,8 @@ def test_solve_balanced(
         rank: int,
         initializer: Optional[Literal["random", "rank2"]],
     ):
-        kwargs = {}
+        # check https://github.com/ott-jax/ott/issues/495
+        kwargs = {"inner_iterations": 10}
         if rank > -1:
             kwargs["initializer"] = initializer
             if initializer == "random":

diff --git a/tests/problems/space/test_mapping_problem.py b/tests/problems/space/test_mapping_problem.py
@@ -110,7 +110,8 @@ def test_solve_balanced(
         initializer: Optional[Literal["random", "rank2"]],
     ):
         adataref, adatasp = _adata_spatial_split(adata_mapping)
-        kwargs = {}
+        # check https://github.com/ott-jax/ott/issues/495
+        kwargs = {"inner_iterations": 10}
         if rank > -1:
             kwargs["initializer"] = initializer
             if initializer == "random":