refactor: move to _coordinax folder

Signed-off-by: nstarman <nstarman@users.noreply.github.com>

pyproject.toml

@@ -195,7 +195,7 @@
     [tool.ruff.lint.isort]
       combine-as-imports = true
       extra-standard-library = ["typing_extensions"]
-      known-first-party = ["dataclassish", "quaxed", "unxt"]
+      known-first-party = ["dataclassish", "quaxed", "unxt", "xmmutablemap"]
       known-local-folder = ["coordinax"]
 
 

src/coordinax/__init__.py

@@ -7,7 +7,8 @@
 
 from jaxtyping import install_import_hook
 
-from . import (
+from . import operators
+from ._coordinax import (
     _base,
     _base_acc,
     _base_pos,
@@ -23,23 +24,22 @@
     _transform,
     _typing,
     _utils,
-    operators,
 )
-from ._base import *
-from ._base_acc import *
-from ._base_pos import *
-from ._base_vel import *
-from ._d1 import *
-from ._d2 import *
-from ._d3 import *
-from ._d4 import *
-from ._dn import *
-from ._exceptions import *
-from ._funcs import *
-from ._space import *
-from ._transform import *
-from ._typing import *
-from ._utils import *
+from ._coordinax._base import *
+from ._coordinax._base_acc import *
+from ._coordinax._base_pos import *
+from ._coordinax._base_vel import *
+from ._coordinax._d1 import *
+from ._coordinax._d2 import *
+from ._coordinax._d3 import *
+from ._coordinax._d4 import *
+from ._coordinax._dn import *
+from ._coordinax._exceptions import *
+from ._coordinax._funcs import *
+from ._coordinax._space import *
+from ._coordinax._transform import *
+from ._coordinax._typing import *
+from ._coordinax._utils import *
 from ._version import version as __version__
 from .setup_package import RUNTIME_TYPECHECKER
 
@@ -62,7 +62,7 @@
 
 # Interoperability
 # Astropy
-from ._interop import coordinax_interop_astropy  # noqa: E402
+from ._coordinax._interop import coordinax_interop_astropy  # noqa: E402
 
 # Runtime Typechecker
 install_import_hook("coordinax", RUNTIME_TYPECHECKER)

src/coordinax/_coordinax/__init__.py

@@ -0,0 +1 @@
+"""Coordinax package."""

src/coordinax/_base.py → src/coordinax/_coordinax/_base.py

@@ -29,8 +29,8 @@
 from dataclassish import field_items, field_values, replace
 from unxt import Quantity, unitsystem
 
+from ._typing import Unit
 from ._utils import classproperty, full_shaped
-from coordinax._typing import Unit
 
 if TYPE_CHECKING:
     from typing_extensions import Self

src/coordinax/_base_pos.py → src/coordinax/_coordinax/_base_pos.py

@@ -19,7 +19,7 @@
 from dataclassish import field_items
 from unxt import Quantity
 
-import coordinax._typing as ct
+from . import _typing as ct
 from ._base import AbstractVector
 from ._mixins import AvalMixin
 from ._utils import classproperty

src/coordinax/_checks.py → src/coordinax/_coordinax/_checks.py

@@ -8,7 +8,7 @@
 import quaxed.array_api as xp
 from unxt import Quantity
 
-from coordinax._typing import BatchableAngle, BatchableLength
+from coordinax._coordinax._typing import BatchableAngle, BatchableLength
 
 _0m = Quantity(0, "meter")
 _0d = Quantity(0, "rad")

src/coordinax/_converters.py → src/coordinax/_coordinax/_converters.py

@@ -5,7 +5,7 @@
 
 from unxt import Quantity
 
-from coordinax._typing import BatchableAngle
+from coordinax._coordinax._typing import BatchableAngle
 
 _2pid = Quantity(360, "deg")
 

src/coordinax/_d1/base.py → src/coordinax/_coordinax/_d1/base.py

@@ -11,11 +11,11 @@
 import quaxed.numpy as jnp
 from unxt import Quantity
 
-from coordinax._base import AbstractVector
-from coordinax._base_acc import AbstractAcceleration
-from coordinax._base_pos import AbstractPosition
-from coordinax._base_vel import AbstractVelocity
-from coordinax._utils import classproperty
+from coordinax._coordinax._base import AbstractVector
+from coordinax._coordinax._base_acc import AbstractAcceleration
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._base_vel import AbstractVelocity
+from coordinax._coordinax._utils import classproperty
 
 
 class AbstractPosition1D(AbstractPosition):

src/coordinax/_d1/cartesian.py → src/coordinax/_coordinax/_d1/cartesian.py

@@ -19,11 +19,11 @@
 from quaxed import lax as qlax
 from unxt import Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration1D, AbstractPosition1D, AbstractVelocity1D
-from coordinax._base_pos import AbstractPosition
-from coordinax._mixins import AvalMixin
-from coordinax._utils import classproperty
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._mixins import AvalMixin
+from coordinax._coordinax._utils import classproperty
 
 
 @final

src/coordinax/_d1/compat.py → src/coordinax/_coordinax/_d1/compat.py

@@ -12,7 +12,7 @@
 
 from .base import AbstractPosition1D
 from .cartesian import CartesianAcceleration1D, CartesianPosition1D, CartesianVelocity1D
-from coordinax._utils import full_shaped
+from coordinax._coordinax._utils import full_shaped
 
 
 @conversion_method(type_from=AbstractPosition1D, type_to=Quantity)  # type: ignore[misc]

src/coordinax/_d1/operate.py → src/coordinax/_coordinax/_d1/operate.py

@@ -11,8 +11,8 @@
 from unxt import Quantity
 
 from .cartesian import CartesianPosition1D
-from coordinax._typing import TimeBatchOrScalar
-from coordinax.operators._base import AbstractOperator, op_call_dispatch
+from coordinax._coordinax._typing import TimeBatchOrScalar
+from coordinax._coordinax.operators._base import AbstractOperator, op_call_dispatch
 
 Q1: TypeAlias = Shaped[Quantity["length"], "*#batch 1"]
 

src/coordinax/_d1/radial.py → src/coordinax/_coordinax/_d1/radial.py

@@ -8,10 +8,10 @@
 
 from unxt import AbstractDistance, Distance, Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration1D, AbstractPosition1D, AbstractVelocity1D
-from coordinax._checks import check_r_non_negative
-from coordinax._utils import classproperty
+from coordinax._coordinax._checks import check_r_non_negative
+from coordinax._coordinax._utils import classproperty
 
 
 @final

src/coordinax/_d1/transform.py → src/coordinax/_coordinax/_d1/transform.py

@@ -9,8 +9,8 @@
 from .base import AbstractAcceleration1D, AbstractPosition1D, AbstractVelocity1D
 from .cartesian import CartesianAcceleration1D, CartesianPosition1D, CartesianVelocity1D
 from .radial import RadialAcceleration, RadialPosition, RadialVelocity
-from coordinax._base_pos import AbstractPosition
-from coordinax._base_vel import AbstractVelocity
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._base_vel import AbstractVelocity
 
 ###############################################################################
 # 1D

src/coordinax/_d2/base.py → src/coordinax/_coordinax/_d2/base.py

@@ -5,11 +5,11 @@
 
 from abc import abstractmethod
 
-from coordinax._base import AbstractVector
-from coordinax._base_acc import AbstractAcceleration
-from coordinax._base_pos import AbstractPosition
-from coordinax._base_vel import AbstractVelocity
-from coordinax._utils import classproperty
+from coordinax._coordinax._base import AbstractVector
+from coordinax._coordinax._base_acc import AbstractAcceleration
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._base_vel import AbstractVelocity
+from coordinax._coordinax._utils import classproperty
 
 
 class AbstractPosition2D(AbstractPosition):

src/coordinax/_d2/cartesian.py → src/coordinax/_coordinax/_d2/cartesian.py

@@ -19,11 +19,11 @@
 from quaxed import lax as qlax
 from unxt import Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration2D, AbstractPosition2D, AbstractVelocity2D
-from coordinax._base_pos import AbstractPosition
-from coordinax._mixins import AvalMixin
-from coordinax._utils import classproperty
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._mixins import AvalMixin
+from coordinax._coordinax._utils import classproperty
 
 
 @final

src/coordinax/_d2/compat.py → src/coordinax/_coordinax/_d2/compat.py

@@ -12,7 +12,7 @@
 
 from .base import AbstractPosition2D
 from .cartesian import CartesianAcceleration2D, CartesianPosition2D, CartesianVelocity2D
-from coordinax._utils import full_shaped
+from coordinax._coordinax._utils import full_shaped
 
 #####################################################################
 # Quantity

src/coordinax/_d2/operate.py → src/coordinax/_coordinax/_d2/operate.py

@@ -11,8 +11,8 @@
 from unxt import Quantity
 
 from .cartesian import CartesianPosition2D
-from coordinax._typing import TimeBatchOrScalar
-from coordinax.operators._base import AbstractOperator, op_call_dispatch
+from coordinax._coordinax._typing import TimeBatchOrScalar
+from coordinax._coordinax.operators._base import AbstractOperator, op_call_dispatch
 
 Q2: TypeAlias = Shaped[Quantity["length"], "*#batch 2"]
 

src/coordinax/_d2/polar.py → src/coordinax/_coordinax/_d2/polar.py

@@ -17,11 +17,11 @@
 from dataclassish import replace
 from unxt import AbstractDistance, Distance, Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration2D, AbstractPosition2D, AbstractVelocity2D
-from coordinax._checks import check_azimuth_range, check_r_non_negative
-from coordinax._converters import converter_azimuth_to_range
-from coordinax._utils import classproperty
+from coordinax._coordinax._checks import check_azimuth_range, check_r_non_negative
+from coordinax._coordinax._converters import converter_azimuth_to_range
+from coordinax._coordinax._utils import classproperty
 
 
 @final

src/coordinax/_d2/transform.py → src/coordinax/_coordinax/_d2/transform.py

@@ -11,7 +11,7 @@
 from .base import AbstractPosition2D, AbstractVelocity2D
 from .cartesian import CartesianAcceleration2D, CartesianPosition2D, CartesianVelocity2D
 from .polar import PolarPosition, PolarVelocity
-from coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._base_pos import AbstractPosition
 
 
 @dispatch

src/coordinax/_d3/base.py → src/coordinax/_coordinax/_d3/base.py

@@ -6,11 +6,11 @@
 from abc import abstractmethod
 from typing_extensions import override
 
-from coordinax._base import AbstractVector
-from coordinax._base_acc import AbstractAcceleration
-from coordinax._base_pos import AbstractPosition
-from coordinax._base_vel import AbstractVelocity
-from coordinax._utils import classproperty
+from coordinax._coordinax._base import AbstractVector
+from coordinax._coordinax._base_acc import AbstractAcceleration
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._base_vel import AbstractVelocity
+from coordinax._coordinax._utils import classproperty
 
 
 class AbstractPosition3D(AbstractPosition):

src/coordinax/_d3/cartesian.py → src/coordinax/_coordinax/_d3/cartesian.py

@@ -20,11 +20,11 @@
 from dataclassish import field_items
 from unxt import Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration3D, AbstractPosition3D, AbstractVelocity3D
-from coordinax._base_pos import AbstractPosition
-from coordinax._mixins import AvalMixin
-from coordinax._utils import classproperty
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._mixins import AvalMixin
+from coordinax._coordinax._utils import classproperty
 
 
 @final

src/coordinax/_d3/compat.py → src/coordinax/_coordinax/_d3/compat.py

@@ -12,7 +12,7 @@
 
 from .base import AbstractPosition3D
 from .cartesian import CartesianAcceleration3D, CartesianPosition3D, CartesianVelocity3D
-from coordinax._utils import full_shaped
+from coordinax._coordinax._utils import full_shaped
 
 #####################################################################
 # Quantity

src/coordinax/_d3/cylindrical.py → src/coordinax/_coordinax/_d3/cylindrical.py

@@ -15,11 +15,11 @@
 import quaxed.array_api as xp
 from unxt import Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration3D, AbstractPosition3D, AbstractVelocity3D
-from coordinax._checks import check_azimuth_range, check_r_non_negative
-from coordinax._converters import converter_azimuth_to_range
-from coordinax._utils import classproperty
+from coordinax._coordinax._checks import check_azimuth_range, check_r_non_negative
+from coordinax._coordinax._converters import converter_azimuth_to_range
+from coordinax._coordinax._utils import classproperty
 
 
 @final

src/coordinax/_d3/operate.py → src/coordinax/_coordinax/_d3/operate.py

@@ -11,8 +11,8 @@
 from unxt import Quantity
 
 from .cartesian import CartesianPosition3D
-from coordinax._typing import TimeBatchOrScalar
-from coordinax.operators._base import AbstractOperator, op_call_dispatch
+from coordinax._coordinax._typing import TimeBatchOrScalar
+from coordinax._coordinax.operators._base import AbstractOperator, op_call_dispatch
 
 Q3: TypeAlias = Shaped[Quantity["length"], "*#batch 3"]
 

src/coordinax/_d3/sphere.py → src/coordinax/_coordinax/_d3/sphere.py

@@ -34,16 +34,16 @@
 from dataclassish import replace
 from unxt import AbstractDistance, AbstractQuantity, Distance, Quantity
 
-import coordinax._typing as ct
+import coordinax._coordinax._typing as ct
 from .base import AbstractAcceleration3D, AbstractPosition3D, AbstractVelocity3D
-from coordinax._base_acc import AbstractAcceleration
-from coordinax._checks import (
+from coordinax._coordinax._base_acc import AbstractAcceleration
+from coordinax._coordinax._checks import (
     check_azimuth_range,
     check_polar_range,
     check_r_non_negative,
 )
-from coordinax._converters import converter_azimuth_to_range
-from coordinax._utils import classproperty
+from coordinax._coordinax._converters import converter_azimuth_to_range
+from coordinax._coordinax._utils import classproperty
 
 _90d = Quantity(90, "deg")
 _180d = Quantity(180, "deg")

src/coordinax/_d3/transform.py → src/coordinax/_coordinax/_d3/transform.py

@@ -22,7 +22,7 @@
     SphericalPosition,
     SphericalVelocity,
 )
-from coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._base_pos import AbstractPosition
 
 ###############################################################################
 # 3D

src/coordinax/_d4/base.py → src/coordinax/_coordinax/_d4/base.py

@@ -6,9 +6,9 @@
 from abc import abstractmethod
 from typing import TYPE_CHECKING
 
-from coordinax._base import AbstractVector
-from coordinax._base_pos import AbstractPosition
-from coordinax._utils import classproperty
+from coordinax._coordinax._base import AbstractVector
+from coordinax._coordinax._base_pos import AbstractPosition
+from coordinax._coordinax._utils import classproperty
 
 if TYPE_CHECKING:
     from typing_extensions import Never

src/coordinax/_d4/operate.py → src/coordinax/_coordinax/_d4/operate.py

@@ -9,7 +9,7 @@
 from unxt import Quantity
 
 from .spacetime import FourVector
-from coordinax.operators._base import AbstractOperator, op_call_dispatch
+from coordinax._coordinax.operators._base import AbstractOperator, op_call_dispatch
 
 
 @op_call_dispatch