Revert "remove kwargs on a more public problem class"

This reverts commit 0b06cc4.
theislab · Sep 22, 2024 · 01c89a2 · 01c89a2
1 parent 376eccf
commit 01c89a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/moscot/problems/cross_modality/_translation.py b/src/moscot/problems/cross_modality/_translation.py
@@ -226,6 +226,7 @@ def solve(  # type: ignore[override]
         threshold: float = 1e-3,
         linear_solver_kwargs: Mapping[str, Any] = types.MappingProxyType({}),
         device: Optional[Literal["cpu", "gpu", "tpu"]] = None,
+        **kwargs: Any,
     ) -> "TranslationProblem[K]":
         r"""Solve the translation problem.
 
@@ -303,6 +304,7 @@ def solve(  # type: ignore[override]
             threshold=threshold,
             linear_solver_kwargs=linear_solver_kwargs,
             device=device,
+            **kwargs,
         )  # type: ignore[return-value]
 
     @property