fix bug in SinkhornProblem (#442)

* fix bug in SinkhornProblem * fix tox.ini
theislab · Jan 4, 2023 · f30b766 · f30b766
1 parent d3f1da6
commit f30b766
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/src/moscot/problems/generic/_generic.py b/src/moscot/problems/generic/_generic.py
@@ -63,21 +63,6 @@ def prepare(
         %(ex_prepare)s
         """
         self.batch_key = key
-        if joint_attr is None:
-            kwargs["xy_callback"] = "local-pca"
-            kwargs.setdefault("xy_callback_kwargs", {})
-        elif isinstance(joint_attr, str):
-            kwargs["xy"] = {
-                "x_attr": "obsm",
-                "x_key": joint_attr,
-                "y_attr": "obsm",
-                "y_key": joint_attr,
-            }
-        elif isinstance(joint_attr, Mapping):
-            kwargs["xy"] = joint_attr
-        else:
-            raise TypeError(f"Unable to interpret `joint_attr` of type `{type(joint_attr)}`.")
-
         xy, kwargs = handle_joint_attr(joint_attr, kwargs)
         xy, _, _ = handle_cost(xy=xy, cost=cost)
         return super().prepare(