🎨 style(dimensions): restructure private files (#372)

Signed-off-by: Nathaniel Starkman <[email protected]>

src/unxt/_src/dimensions/__init__.py

@@ -6,4 +6,8 @@
 
 __all__ = ["AbstractDimension", "dimension", "dimension_of"]
 
-from .core import AbstractDimension, dimension, dimension_of
+from .api import AbstractDimension, dimension, dimension_of
+
+# Register the dispatches
+# isort: split
+from . import core  # noqa: F401

src/unxt/_src/dimensions/api.py

@@ -0,0 +1,41 @@
+"""Units objects in JAX.
+
+Copyright (c) 2023 Galactic Dynamics. All rights reserved.
+"""
+
+__all__ = ["dimension", "dimension_of"]
+
+from typing import Any, TypeAlias
+
+import astropy.units as apyu
+from plum import dispatch
+
+AbstractDimension: TypeAlias = apyu.PhysicalType
+
+
+@dispatch.abstract
+def dimension(obj: Any, /) -> AbstractDimension:
+    """Construct the dimension.
+
+    .. note::
+
+        This function uses multiple dispatch. Dispatches made in other modules
+        may not be included in the rendered docs. To see the full range of
+        options, execute ``unxt.dims.dimension.methods`` in an interactive
+        Python session.
+
+    """
+
+
+@dispatch.abstract
+def dimension_of(obj: Any, /) -> AbstractDimension:
+    """Return the dimension of the given units.
+
+    .. note::
+
+        This function uses multiple dispatch. Dispatches made in other modules
+        may not be included in the rendered docs. To see the full range of
+        options, execute ``unxt.dimension_of.methods`` in an interactive Python
+        session.
+
+    """

src/unxt/_src/dimensions/core.py

@@ -3,36 +3,23 @@
 Copyright (c) 2023 Galactic Dynamics. All rights reserved.
 """
 
-__all__ = ["dimension", "dimension_of"]
+__all__: list[str] = []
 
 from typing import Any, TypeAlias
 
 import astropy.units as apyu
 from astropy.units import Unit
 from plum import dispatch
 
+from .api import AbstractDimension
+
 AbstractUnits: TypeAlias = apyu.UnitBase | Unit
-AbstractDimension: TypeAlias = apyu.PhysicalType
 
 
 # ===================================================================
 # Construct the dimensions
 
 
-@dispatch.abstract
-def dimension(obj: Any, /) -> AbstractDimension:
-    """Construct the dimension.
-
-    .. note::
-
-        This function uses multiple dispatch. Dispatches made in other modules
-        may not be included in the rendered docs. To see the full range of
-        options, execute ``unxt.dims.dimension.methods`` in an interactive
-        Python session.
-
-    """
-
-
 @dispatch
 def dimension(obj: AbstractDimension, /) -> AbstractDimension:
     """Construct dimension from a dimension object.
@@ -71,20 +58,6 @@ def dimension(obj: str, /) -> AbstractDimension:
 # Get the dimension
 
 
-@dispatch.abstract
-def dimension_of(obj: Any, /) -> AbstractDimension:
-    """Return the dimension of the given units.
-
-    .. note::
-
-        This function uses multiple dispatch. Dispatches made in other modules
-        may not be included in the rendered docs. To see the full range of
-        options, execute ``unxt.dimension_of.methods`` in an interactive Python
-        session.
-
-    """
-
-
 @dispatch
 def dimension_of(obj: Any, /) -> None:
     """Most objects have no dimension.