fix alpha tests

src/moscot/backends/ott/solver.py

@@ -451,8 +451,6 @@ def _prepare(
             geom_kwargs["cost_matrix_rank"] = cost_matrix_rank
         geom_xx = self._create_geometry(x, t=time_scales_heat_kernel.x, is_linear_term=False, **geom_kwargs)
         geom_yy = self._create_geometry(y, t=time_scales_heat_kernel.y, is_linear_term=False, **geom_kwargs)
-        if alpha is None:
-            alpha = 1.0 if xy is None else 0.5  # set defaults according to the data provided
         if alpha <= 0.0:
             raise ValueError(f"Expected `alpha` to be in interval `(0, 1]`, found `{alpha}`.")
         if (alpha == 1.0 and xy is not None) or (alpha != 1.0 and xy is None):

tests/problems/base/test_general_problem.py

@@ -288,7 +288,7 @@ def test_set_graph_x_y(self, adata_x: AnnData, adata_y: AnnData, ts: Tuple[Optio
         assert ta2.tag == Tag.GRAPH
         assert ta2.cost == "geodesic"
 
-        prob1 = prob1.solve(epsilon=10.0)
+        prob1 = prob1.solve(epsilon=10.0, alpha=1.0)
 
         prob2 = OTProblem(adata_x, adata_y)
         prob2 = prob2.prepare(
@@ -313,7 +313,7 @@ def test_set_graph_x_y(self, adata_x: AnnData, adata_y: AnnData, ts: Tuple[Optio
         assert ta2.tag == Tag.GRAPH
         assert ta2.cost == "geodesic"
 
-        prob2 = prob2.solve(epsilon=10.0)
+        prob2 = prob2.solve(epsilon=10.0, alpha=1.0)
 
         assert not np.allclose(prob1.solution._output.geom.cost_matrix, prob2.solution._output.geom.cost_matrix)
 

tests/problems/time/test_temporal_base_problem.py

@@ -111,7 +111,7 @@ def test_posterior_growth_rates(self, adata_time_marginal_estimations: AnnData):
             b=True,
             marginal_kwargs={"proliferation_key": "proliferation"},
         )
-        prob = prob.solve(max_iterations=10)
+        prob = prob.solve(max_iterations=10, alpha=1.0)
         assert prob.delta == (t2 - t1)
 
         gr = prob.posterior_growth_rates