fix some bug

theislab · Nov 25, 2024 · 90fa16d · 90fa16d
1 parent be8460a
commit 90fa16d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/moscot/problems/space/_mixins.py b/src/moscot/problems/space/_mixins.py
@@ -30,13 +30,13 @@
 from moscot._types import ArrayLike, Device_t, Str_Dict_t
 from moscot.base.problems._mixins import AnalysisMixin
 from moscot.base.problems.compound_problem import B, K
-from moscot.base.problems.problem import AbstractSpSc, AbstractSrcTgt
+from moscot.base.problems.problem import AbstractSpSc
 from moscot.utils.subset_policy import StarPolicy
 
 __all__ = ["SpatialAlignmentMixin", "SpatialMappingMixin"]
 
 
-class SpatialAlignmentMixin(AnalysisMixin[K, B], AbstractSrcTgt):
+class SpatialAlignmentMixin(AnalysisMixin[K, B]):
     """Spatial alignment mixin class."""
 
     def __init__(self, *args: Any, **kwargs: Any) -> None: