transformnd 0.1.0__tar.gz → 0.2.1__tar.gz

{transformnd-0.1.0 → transformnd-0.2.1}/PKG-INFO

@@ -1,10 +1,21 @@
 Metadata-Version: 2.4
 Name: transformnd
-Version: 0.1.0
+Version: 0.2.1
 Summary: ND coordinate transformations
 Author: Chris Barnes
 Author-email: Chris Barnes <chris.barnes@gerbi-gmb.de>
 License-Expression: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: GIS
+Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: numpy>=2
 Requires-Dist: networkx>=3
 Requires-Dist: array-api-compat>=1.14
@@ -15,6 +26,9 @@ Requires-Dist: polars>=1.40.1 ; extra == 'polars'
 Requires-Dist: shapely>=2.1.2 ; extra == 'shapely'
 Requires-Dist: morphops>=0.1.13 ; extra == 'thinplatesplines'
 Requires-Python: >=3.12, <4.0
+Project-URL: documentation, https://transformnd.readthedocs.io/en/latest/
+Project-URL: source, https://github.com/clbarnes/transformnd
+Project-URL: issues, https://github.com/clbarnes/transformnd/issues
 Provides-Extra: movingleastsquares
 Provides-Extra: pandas
 Provides-Extra: polars
@@ -24,10 +38,10 @@ Description-Content-Type: text/markdown
 
 # transformnd
 
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-[![GitHub](https://img.shields.io/github/license/clbarnes/transformnd)](https://github.com/clbarnes/transformnd/blob/main/LICENSE)
-[![Test Status](https://img.shields.io/github/workflow/status/clbarnes/transformnd/ci)](https://github.com/clbarnes/transformnd/actions/workflows/ci.yaml)
-[![Docs Status](https://img.shields.io/github/workflow/status/clbarnes/transformnd/docs?label=docs)](https://clbarnes.github.io/transformnd/)
+[![PyPI - Version](https://img.shields.io/pypi/v/transformnd)](https://pypi.org/project/transformnd/)
+[![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/clbarnes/transformnd/ci.yaml)](https://github.com/clbarnes/transformnd/actions)
+[![Read the Docs](https://img.shields.io/readthedocs/transformnd)](https://transformnd.readthedocs.io)
+![PyPI - License](https://img.shields.io/pypi/l/transformnd)
 
 A library providing an API for coordinate transformations,
 as well as some common transforms.
@@ -45,7 +59,7 @@ co-developed with [xform](https://github.com/schlegelp/xform/) as a red team pro
 `Transform` subclasses which are restricted to certain dimensionalities
 can specify this in their `ndim` class variable.
 Instances of `Transform` subclasses can further restrict their `ndim`.
-Use `self._validate_coords(coords)` in `__call__` to ensure the coordinates
+Use `self._validate_coords(coords)` in the `apply` method to ensure the coordinates
 are of valid type and dimensions.
 
 Additionally, `transformnd` provides an interface for transforming types other than NxD numpy arrays,
@@ -91,27 +105,25 @@ Contributions of additional transforms and adapters are welcome!
 Even if they're only thin wrappers around an external library,
 the downstream ecosystem benefits from a consistent API.
 
-Such external transformation libraries should be specified as "extras",
+Such external transformation libraries should be specified as "extras" (`pyproject.toml:project.optional-dependencies`),
 and be contained in a submodule so that they are not immediately imported
 with `transformnd`.
-Dependencies for new adapters do not need to be included in `transformnd`'s dependencies,
-but should be specified in the `requirements.txt` for tests.
 
 Alternatively, consider adopting `transformnd`'s base classes in your own library,
 and have your transformation instantly compatible for downstream users.
 
 Methods which MUST be implemented:
 
-- `__init__`: should validate parameters and set `self.ndim` if the parameters constrain the dimensionality
+- `__init__`: should validate parameters and must call the `super()` constructor
 - `apply`: should call `_validate_coords` method early to check that the given coordinates are the correct shape
 
 Methods which SHOULD be implemented if applicable:
 
 - `to_device`: if any of the transformation's parameters need to be placed on a specific device (e.g. affine matrices on the GPU)
 - `is_identity`: if you can cheaply check whether your transformation is an identity transformation. The base class implementation returns `False`.
-- `into_affine`: if your transformation can be represented as an affine matrix. The base class implementation returns `None`.
+- `to_affine`: if your transformation can be represented as an affine matrix. The base class implementation returns `None`.
 - `invert`: if your transformation can be inverted (default None if not)
-  - This automatically implements `__invert__` (the `~my_transform` operator), which raises NotImplemented if `invert` would return `None`.
+  - This automatically implements `__invert__` (the `~my_transform` operator), which returns `NotImplemented` (probably raising `NotImplementedError`) if `invert` would return `None`.
 
 ## Contributing
 
@@ -121,7 +133,8 @@ Methods which SHOULD be implemented if applicable:
   - `prek install-hooks && prek run --all-files` to get started.
 - Use [`just`](https://github.com/casey/just) for common development tasks (format, lint, test, generate docs, run benchmarks).
   - `just` to list commands.
-- Docs are generated with `pdoc` (use `just doc`) and hosted on ReadTheDocs
+- Docs are generated with [`pdoc`](https://pdoc.dev/) (use `just doc`) and hosted on ReadTheDocs
+- `just bump` bumps the version, commits, and tags (but does not push); depends on [`schpet/changelog`](https://github.com/schpet/changelog)
 
 ## Thanks
 

{transformnd-0.1.0 → transformnd-0.2.1}/README.md

@@ -1,9 +1,9 @@
 # transformnd
 
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-[![GitHub](https://img.shields.io/github/license/clbarnes/transformnd)](https://github.com/clbarnes/transformnd/blob/main/LICENSE)
-[![Test Status](https://img.shields.io/github/workflow/status/clbarnes/transformnd/ci)](https://github.com/clbarnes/transformnd/actions/workflows/ci.yaml)
-[![Docs Status](https://img.shields.io/github/workflow/status/clbarnes/transformnd/docs?label=docs)](https://clbarnes.github.io/transformnd/)
+[![PyPI - Version](https://img.shields.io/pypi/v/transformnd)](https://pypi.org/project/transformnd/)
+[![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/clbarnes/transformnd/ci.yaml)](https://github.com/clbarnes/transformnd/actions)
+[![Read the Docs](https://img.shields.io/readthedocs/transformnd)](https://transformnd.readthedocs.io)
+![PyPI - License](https://img.shields.io/pypi/l/transformnd)
 
 A library providing an API for coordinate transformations,
 as well as some common transforms.
@@ -21,7 +21,7 @@ co-developed with [xform](https://github.com/schlegelp/xform/) as a red team pro
 `Transform` subclasses which are restricted to certain dimensionalities
 can specify this in their `ndim` class variable.
 Instances of `Transform` subclasses can further restrict their `ndim`.
-Use `self._validate_coords(coords)` in `__call__` to ensure the coordinates
+Use `self._validate_coords(coords)` in the `apply` method to ensure the coordinates
 are of valid type and dimensions.
 
 Additionally, `transformnd` provides an interface for transforming types other than NxD numpy arrays,
@@ -67,27 +67,25 @@ Contributions of additional transforms and adapters are welcome!
 Even if they're only thin wrappers around an external library,
 the downstream ecosystem benefits from a consistent API.
 
-Such external transformation libraries should be specified as "extras",
+Such external transformation libraries should be specified as "extras" (`pyproject.toml:project.optional-dependencies`),
 and be contained in a submodule so that they are not immediately imported
 with `transformnd`.
-Dependencies for new adapters do not need to be included in `transformnd`'s dependencies,
-but should be specified in the `requirements.txt` for tests.
 
 Alternatively, consider adopting `transformnd`'s base classes in your own library,
 and have your transformation instantly compatible for downstream users.
 
 Methods which MUST be implemented:
 
-- `__init__`: should validate parameters and set `self.ndim` if the parameters constrain the dimensionality
+- `__init__`: should validate parameters and must call the `super()` constructor
 - `apply`: should call `_validate_coords` method early to check that the given coordinates are the correct shape
 
 Methods which SHOULD be implemented if applicable:
 
 - `to_device`: if any of the transformation's parameters need to be placed on a specific device (e.g. affine matrices on the GPU)
 - `is_identity`: if you can cheaply check whether your transformation is an identity transformation. The base class implementation returns `False`.
-- `into_affine`: if your transformation can be represented as an affine matrix. The base class implementation returns `None`.
+- `to_affine`: if your transformation can be represented as an affine matrix. The base class implementation returns `None`.
 - `invert`: if your transformation can be inverted (default None if not)
-  - This automatically implements `__invert__` (the `~my_transform` operator), which raises NotImplemented if `invert` would return `None`.
+  - This automatically implements `__invert__` (the `~my_transform` operator), which returns `NotImplemented` (probably raising `NotImplementedError`) if `invert` would return `None`.
 
 ## Contributing
 
@@ -97,7 +95,8 @@ Methods which SHOULD be implemented if applicable:
   - `prek install-hooks && prek run --all-files` to get started.
 - Use [`just`](https://github.com/casey/just) for common development tasks (format, lint, test, generate docs, run benchmarks).
   - `just` to list commands.
-- Docs are generated with `pdoc` (use `just doc`) and hosted on ReadTheDocs
+- Docs are generated with [`pdoc`](https://pdoc.dev/) (use `just doc`) and hosted on ReadTheDocs
+- `just bump` bumps the version, commits, and tags (but does not push); depends on [`schpet/changelog`](https://github.com/schpet/changelog)
 
 ## Thanks
 

{transformnd-0.1.0 → transformnd-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "transformnd"
-version = "0.1.0"
+version = "0.2.1"
 description = "ND coordinate transformations"
 readme = "README.md"
 authors = [{ name = "Chris Barnes", email = "chris.barnes@gerbi-gmb.de" }]
@@ -12,6 +12,24 @@ dependencies = [
     "typing-extensions>=4.15.0",
 ]
 license = "MIT"
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Topic :: Scientific/Engineering",
+    "Topic :: Scientific/Engineering :: GIS",
+    "Topic :: Scientific/Engineering :: Bio-Informatics",
+    "Topic :: Scientific/Engineering :: Image Processing",
+    "Natural Language :: English",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+]
+
+[project.urls]
+documentation = "https://transformnd.readthedocs.io/en/latest/"
+source = "https://github.com/clbarnes/transformnd"
+issues = "https://github.com/clbarnes/transformnd/issues"
 
 [project.optional-dependencies]
 thinplatesplines = [

{transformnd-0.1.0 → transformnd-0.2.1}/src/transformnd/__init__.py

@@ -3,7 +3,8 @@
 """
 
 from .base import Transform, TransformSequence, TransformWrapper
-from .util import SpaceRef, TransformSignature, check_ndim
+from .util import SpaceRef
+from .types import Spaces, TransformSignature, NDims
 from . import transforms
 from . import adapters
 from .graph import TransformGraph
@@ -18,7 +19,8 @@ __all__ = [
     "TransformWrapper",
     "TransformSignature",
     "SpaceRef",
-    "check_ndim",
     "transforms",
     "adapters",
+    "Spaces",
+    "NDims",
 ]

{transformnd-0.1.0 → transformnd-0.2.1}/src/transformnd/base.py

@@ -11,17 +11,13 @@ from array_api_compat import array_namespace
 
 from .util import (
     SpaceRef,
-    TransformSignature,
-    check_ndim,
-    dim_intersection,
-    invert_spaces,
     same_or_none,
     space_str,
-    to_single_ndim,
-    window,
-    SpaceTuple,
     ArrayT,
 )
+from itertools import pairwise
+
+from .types import TransformSignature, Spaces, NDims
 
 if TYPE_CHECKING:
     from .transforms import Affine
@@ -30,41 +26,28 @@ if TYPE_CHECKING:
 class Transform[ArrayT](ABC):
     """Base class for transforms."""
 
-    ndim: set[int] | None = None
-
     def __init__(
         self,
+        ndims: NDims,
         *,
-        spaces: SpaceTuple = (None, None),
+        spaces: Spaces = Spaces(None, None),
     ):
         """
         Parameters
         ----------
-        spaces : tuple[SpaceRef, SpaceRef]
+        spaces
             Optional source and target spaces
         """
-        self.spaces = spaces
-
-    @property
-    def source_space(self):
-        return self.spaces[0]
-
-    @property
-    def target_space(self):
-        return self.spaces[1]
+        self.ndims: NDims = ndims
+        self.spaces: Spaces = spaces
 
     def is_identity(self) -> bool:
         """Whether this is a no-op transformation."""
         return False
 
-    def to_affine(self, ndim: int | None = None) -> Affine[ArrayT] | None:
+    def to_affine(self) -> Affine[ArrayT] | None:
         """Convert the transform into affine, if conversion is possible.
 
-        Parameters
-        ----------
-        dim: int, optional
-            Total number of dimensions; If None, dim is set equal to self.ndim.
-
         Returns
         -------
         Transform | None:
@@ -74,9 +57,9 @@ class Transform[ArrayT](ABC):
         return None
 
     def _validate_coords(self, coords: ArrayT) -> ArrayT:
-        """Check that dimension of coords are supported.
+        """Check that input coordinates are of the correct shape.
 
-        Also ensure that coords is a 2D array.
+        Also ensure that coords is a 2Darray.
 
         Parameters
         ----------
@@ -91,7 +74,11 @@ class Transform[ArrayT](ABC):
         xp = array_namespace(coords)
         if xp.ndim(coords) != 2:
             raise ValueError("Coords must be a 2D array")
-        check_ndim(xp.shape(coords)[1], self.ndim)
+        dim = xp.shape(coords)[1]
+        if xp.shape(coords)[1] != self.ndims.source:
+            raise ValueError(
+                f"Coords must have dimensionality {self.ndims.source}, got {dim}"
+            )
         return coords
 
     @abstractmethod
@@ -112,8 +99,6 @@ class Transform[ArrayT](ABC):
 
     def invert(self) -> Transform | None:
         """Invert the transformation, returning `None` if not possible."""
-        if self.is_identity():
-            return copy(self)
         return None
 
     def __invert__(self) -> Transform:
@@ -168,10 +153,10 @@ class Transform[ArrayT](ABC):
         """
         if not isinstance(other, Transform):
             return NotImplemented
-        transforms = get_transform_list(self) + get_transform_list(other)
+        transforms = as_transform_list(self) + as_transform_list(other)
         return TransformSequence[ArrayT](
             transforms,
-            spaces=(self.source_space, other.target_space),
+            spaces=Spaces(self.spaces.source, other.spaces.target),
         )
 
     def __ror__(self, other: Transform[ArrayT]) -> TransformSequence[ArrayT]:
@@ -189,16 +174,16 @@ class Transform[ArrayT](ABC):
         """
         if not isinstance(other, Transform):
             return NotImplemented
-        transforms = get_transform_list(other) + get_transform_list(self)
+        transforms = as_transform_list(other) + as_transform_list(self)
         return TransformSequence(
             transforms,
-            spaces=(other.source_space, self.target_space),
+            spaces=Spaces(other.spaces.source, self.spaces.target),
         )
 
     def __str__(self) -> str:
         cls_name = type(self).__name__
-        src = space_str(self.source_space)
-        tgt = space_str(self.target_space)
+        src = space_str(self.spaces.source)
+        tgt = space_str(self.spaces.target)
         return f"{cls_name}[{src}->{tgt}]"
 
 
@@ -208,9 +193,10 @@ class TransformWrapper(Transform[ArrayT]):
     def __init__(
         self,
         fn: TransformSignature[ArrayT],
-        ndim: set[int] | int | None = None,
+        in_ndim: int,
+        out_ndim: int,
         *,
-        spaces: SpaceTuple = (None, None),
+        spaces: Spaces = Spaces(None, None),
     ):
         """Wrapper around an arbitrary function.
 
@@ -224,13 +210,8 @@ class TransformWrapper(Transform[ArrayT]):
         spaces : tuple[SpaceRef, SpaceRef]
             Optional source and target spaces
         """
-        super().__init__(spaces=spaces)
+        super().__init__(NDims(in_ndim, out_ndim), spaces=spaces)
         self.fn = fn
-        if ndim is not None:
-            if isinstance(ndim, int):
-                self.ndim = {ndim}
-            else:
-                self.ndim = set(ndim)
 
     def apply(self, coords: ArrayT) -> ArrayT:
         self._validate_coords(coords)
@@ -242,12 +223,12 @@ def _with_spaces(
     source_space: SpaceRef | None = None,
     target_space: SpaceRef | None = None,
 ) -> Transform[ArrayT]:
-    src_tgt = (t.source_space, t.target_space)
+    src_tgt = (t.spaces.source, t.spaces.target)
     src = same_or_none(src_tgt[0], source_space, default=None)
     tgt = same_or_none(src_tgt[1], target_space, default=None)
     if (src, tgt) != src_tgt:
         t = copy(t)
-        t.spaces = (src, tgt)
+        t.spaces = Spaces(src, tgt)
     return t
 
 
@@ -256,9 +237,9 @@ def infer_spaces(
 ) -> list[Transform[ArrayT]]:
     prev_tgts = [source_space]
     next_srcs = []
-    for t1, t2 in window(transforms, 2):
-        prev_tgts.append(t1.target_space)
-        next_srcs.append(t2.source_space)
+    for t1, t2 in pairwise(transforms):
+        prev_tgts.append(t1.spaces.target)
+        next_srcs.append(t2.spaces.source)
 
     next_srcs.append(target_space)
 
@@ -268,7 +249,7 @@ def infer_spaces(
     return out
 
 
-def get_transform_list(t: Transform[ArrayT]) -> list[Transform[ArrayT]]:
+def as_transform_list(t: Transform[ArrayT]) -> list[Transform[ArrayT]]:
     if isinstance(t, TransformSequence):
         return t.transforms.copy()
     else:
@@ -282,7 +263,7 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
         self,
         transforms: Sequence[Transform[ArrayT]],
         *,
-        spaces: SpaceTuple = (None, None),
+        spaces: Spaces = Spaces(None, None),
     ) -> None:
         """Combine transforms by chaining them.
 
@@ -291,10 +272,10 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
 
         Parameters
         ----------
-        transforms : List[Transform[ArrayT]]
+        transforms :
             Items which are a TransformSequences
             will each still be treated as a single transform.
-        spaces : tuple[SpaceRef, SpaceRef]
+        spaces :
             Optional source and target spaces.
             Can also be inferred from the first and last transforms.
 
@@ -304,23 +285,19 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
             If spaces are incompatible.
         """
         ts = infer_spaces(transforms, *spaces)
+        if not ts:
+            raise ValueError("Empty transform sequence")
+
+        spaces = Spaces(ts[0].spaces.source, ts[-1].spaces.target)
+        ndims = NDims(ts[0].ndims.source, ts[-1].ndims.target)
 
         super().__init__(
-            spaces=(
-                ts[0].source_space,
-                ts[-1].target_space,
-            ),
+            ndims,
+            spaces=spaces,
         )
 
         self.transforms: list[Transform[ArrayT]] = ts
 
-        self.ndim = None
-        for t in self.transforms:
-            self.ndim = dim_intersection(self.ndim, t.ndim)
-
-        if self.ndim is not None and len(self.ndim) == 0:
-            raise ValueError("Transforms have incompatible dimensionalities")
-
     def __iter__(self) -> Iterator[Transform[ArrayT]]:
         """Iterate through component transforms.
 
@@ -346,7 +323,7 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
             return None
         return type(self)(
             transforms,
-            spaces=invert_spaces(self.spaces),
+            spaces=self.spaces.invert(),
         )
 
     def apply(self, coords: ArrayT) -> ArrayT:
@@ -371,7 +348,7 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
         -------
         List[SpaceRef]
         """
-        spaces = [self.source_space] + [t.target_space for t in self.transforms]
+        spaces = [self.spaces.source] + [t.spaces.target for t in self.transforms]
         if skip_none:
             spaces = [s for s in spaces if s is not None]
         return spaces
@@ -389,7 +366,7 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
     def is_identity(self) -> bool:
         return all(t.is_identity() for t in self)
 
-    def simplify(self, ndim: int | None = None, drop_inverse: bool = False):
+    def simplify(self, drop_inverse: bool = True):
         """Reduce the number of transformations in this sequence if possible.
 
         - Compose consecutive transformations which can be expressed as affines
@@ -403,15 +380,15 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
         e.g. `Translation(1) | Translation(-1)`.
         """
         from .transforms.bijection import Bijection
+        from .transforms import Identity
 
-        ndim = to_single_ndim(ndim, self.ndim)
         out: list[Transform[ArrayT]] = []
         affine = None
         for t in self.transforms:
             if drop_inverse and isinstance(t, Bijection):
                 t = t.forward
 
-            new_affine = t.to_affine(ndim)
+            new_affine = t.to_affine()
 
             if new_affine is None:
                 if affine is not None:
@@ -428,7 +405,16 @@ class TransformSequence(Transform[ArrayT], Sequence[Transform[ArrayT]]):
         if affine is not None:
             add_to_output(affine, out)
 
-        return type(self)(out)
+        if not out:
+            out.append(Identity(self.ndims.source))
+
+        return type(self)(out, spaces=self.spaces)
+
+    def to_affine(self) -> Affine[ArrayT] | None:
+        simple = self.simplify(True)
+        if len(simple) != 1:
+            return None
+        return simple[0].to_affine()
 
 
 def add_to_output(transform: Transform, lst: list[Transform]) -> bool:
@@ -436,6 +422,6 @@ def add_to_output(transform: Transform, lst: list[Transform]) -> bool:
         return False
 
     transform = copy(transform)
-    transform.spaces = (None, None)
+    transform.spaces = Spaces(None, None)
     lst.append(transform)
     return True

transformnd-0.2.1/src/transformnd/constants.py

@@ -0,0 +1 @@
+UNSPECIFIED_SPACE_NAME = "???"

{transformnd-0.1.0 → transformnd-0.2.1}/src/transformnd/extents/base.py

@@ -4,7 +4,8 @@ from abc import ABC, abstractmethod
 class Extents[ArrayT](ABC):
     """Base class for determining whether coordinates are "inside" a space."""
 
-    ndim: set[int] | None = None
+    def __init__(self, ndim: int) -> None:
+        self.ndim = ndim
 
     @abstractmethod
     def contains(self, coords: ArrayT) -> ArrayT: ...

{transformnd-0.1.0 → transformnd-0.2.1}/src/transformnd/extents/bounding_box.py

@@ -1,6 +1,6 @@
 from functools import lru_cache
 from array_api_compat import array_namespace
-from ..util import ArrayT, are_coords
+from ..util import ArrayT
 from .base import Extents
 from array_api_compat import device as xp_device
 
@@ -13,9 +13,9 @@ class BoundingBox(Extents[ArrayT]):
         if len(xp.shape(mins)) != 1:
             raise ValueError("mins and maxes must be 1D")
 
-        self.ndim = {xp.shape(mins)[0]}
         self.mins = mins
         self.maxes = maxes
+        super().__init__(xp.shape(mins)[0])
 
     @lru_cache()
     def extents_cast(self, namespace, device) -> tuple[ArrayT, ArrayT]:
@@ -25,7 +25,13 @@ class BoundingBox(Extents[ArrayT]):
         )
 
     def _validate_coords(self, coords: ArrayT) -> ArrayT:
-        return are_coords(coords, self.ndim)
+        xp = array_namespace(coords)
+        if xp.ndim(coords) != 2:
+            raise ValueError("Coords must be a 2D array")
+        dim = xp.shape(coords)[1]
+        if xp.shape(coords)[1] != self.ndim:
+            raise ValueError(f"Coords must have dimensionality {self.ndim}, got {dim}")
+        return coords
 
     def contains(self, coords: ArrayT) -> ArrayT:
         coords = self._validate_coords(coords)