fix test

theislab · Dec 10, 2024 · f15c4e3 · f15c4e3
1 parent 97846e2
commit f15c4e3
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/backends/ott/test_backend.py b/tests/backends/ott/test_backend.py
@@ -350,10 +350,7 @@ def test_pull(
         b, ndim = (b, b.shape[1]) if batched else (b[:, 0], None)
         xx, yy = xy
         solver = solver_t()
-        additional_kwargs = {"alpha": 1.0} if xy is None else {}
-        out = solver(
-            a=jnp.ones(len(x)) / len(x), b=jnp.ones(len(y)) / len(y), x=x, y=y, xy=(xx, yy), **additional_kwargs
-        )
+        out = solver(a=jnp.ones(len(x)) / len(x), b=jnp.ones(len(y)) / len(y), x=x, y=y, xy=(xx, yy), alpha=0.5)
         p = out.pull(b, scale_by_marginals=False)
 
         assert isinstance(out, BaseDiscreteSolverOutput)