fix bug in quad problem marginals (#666)

Co-authored-by: Dominik Klein <[email protected]>
theislab · Mar 13, 2024 · 6ea9a16 · 6ea9a16
1 parent 647477f
commit 6ea9a16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/moscot/backends/ott/solver.py b/src/moscot/backends/ott/solver.py
@@ -418,7 +418,7 @@ def _prepare(
             check_shapes(geom_xx, geom_yy, geom_xy)
 
         self._problem = quadratic_problem.QuadraticProblem(
-            geom_xx, geom_yy, geom_xy, fused_penalty=fused_penalty, **kwargs
+            geom_xx, geom_yy, geom_xy, fused_penalty=fused_penalty, a=self._a, b=self._b, **kwargs
         )
         return self._problem