🎉 init: coordinax_astro

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

Author: nstarman

.github/workflows/ci.yml

@@ -69,11 +69,11 @@ jobs:
         run: uv python install ${{ matrix.python-version }}
 
       - name: Install the project
-        run: uv sync --group test
+        run: uv sync --all-packages --group test --active
 
       - name: Test package
         run: >-
-          uv run pytest -ra --cov --cov-report=xml --cov-report=term
+          uv run --frozen pytest --cov --cov-report=xml --cov-report=term
           --durations=20 --arraydiff -m"not slow"
 
       - name: Upload coverage report
@@ -108,8 +108,8 @@ jobs:
 
       - name: Test package
         run: >-
-          uv run pytest -ra --cov --cov-report=xml --cov-report=term
-          --durations=20
+          uv run --active --frozen pytest --cov --cov-report=xml
+          --cov-report=term --durations=20
 
       - name: Upload coverage report
         uses: codecov/codecov-action@v5.4.3

docs/guides/vectors.md

@@ -68,7 +68,7 @@ CartesianVel3D(
   {class}`~coordinax.vecs.CartesianPosND`
 - Similar classes for velocities (`CartesianVel*`), accelerations
   (`CartesianAcc*`), etc.
-- Spacetime vectors {class}`~coordinax.vecs.FourVector`
+- Spacetime vectors {class}`~coordinax_astro.FourVector`
 
 ## Conversion Between Representations
 

noxfile.py

@@ -94,7 +94,7 @@ def docs(session: nox.Session) -> None:
     if args.builder != "html" and args.serve:
         session.error("Must not specify non-HTML builder with --serve")
 
-    session.run("uv", "sync", "--group", "docs", "--active")
+    session.run("uv", "sync", "--all-packages", "--group", "docs", "--active")
     session.chdir("docs")
 
     if args.builder == "linkcheck":

packages/coordinax-astro/README.md

@@ -0,0 +1,32 @@
+[project]
+name = "coordinax_astro"
+dynamic = ["version"]
+description = "Astronomical coordinates in JAX"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.11"
+authors = [
+    { name = "Coordinax Devs", email = "nstarman@users.noreply.github.com" },
+    { name = "Nathaniel Starkman", email = "nstarman@users.noreply.github.com" },
+    { name = "Adrian Price-Whelan", email = "adrn@users.noreply.github.com" },
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: Developers",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Scientific/Engineering",
+    "Typing :: Typed",
+]
+dependencies = [
+    "coordinax",
+]
+
+[tool.uv.sources]
+coordinax = { workspace = true }

packages/coordinax-astro/pyproject.toml

@@ -0,0 +1,7 @@
+"""Coordinax for Astronomy."""
+
+__all__ = [
+    "FourVector",
+]
+
+from ._src import FourVector

packages/coordinax-astro/src/coordinax_astro/__init__.py

@@ -0,0 +1,7 @@
+"""Frames for Astronomy."""
+
+__all__ = [
+    "FourVector",
+]
+
+from .vecs import FourVector

packages/coordinax-astro/src/coordinax_astro/_src/__init__.py

@@ -0,0 +1,15 @@
+"""Frames for Astronomy."""
+
+__all__ = [
+    "FourVector",
+]
+
+from .spacetime import FourVector
+
+# Register by importing
+# isort: split
+from . import (
+    register_convert,  # noqa: F401
+    register_primitives,  # noqa: F401
+    register_vectorapi,  # noqa: F401
+)

packages/coordinax-astro/src/coordinax_astro/_src/vecs/__init__.py

@@ -11,14 +11,8 @@
 import unxt as u
 from unxt.quantity import BareQuantity
 
+import coordinax.vecs as cxv
 from .spacetime import FourVector
-from coordinax._src.vectors.d3 import (
-    CartesianPos3D,
-    CylindricalPos,
-    LonLatSphericalPos,
-    MathSphericalPos,
-    SphericalPos,
-)
 
 
 @conversion_method(type_from=FourVector, type_to=BareQuantity)
@@ -77,8 +71,8 @@ def fourvec_to_quantity(obj: FourVector, /) -> Shaped[u.Quantity["length"], "*ba
     return jnp.concat([ct, cart], axis=-1)
 
 
-@conversion_method(type_from=FourVector, type_to=CartesianPos3D)  # type: ignore[arg-type]
-def convert_4vec_to_cart3d(obj: FourVector, /) -> CartesianPos3D:
+@conversion_method(type_from=FourVector, type_to=cxv.CartesianPos3D)  # type: ignore[arg-type]
+def convert_4vec_to_cart3d(obj: FourVector, /) -> cxv.CartesianPos3D:
     """Convert a 4-vector to a Cartesian 3-vector.
 
     Examples
@@ -93,11 +87,11 @@ def convert_4vec_to_cart3d(obj: FourVector, /) -> CartesianPos3D:
         [1 2 3]>
 
     """
-    return convert(obj.q, CartesianPos3D)
+    return convert(obj.q, cxv.CartesianPos3D)
 
 
-@conversion_method(type_from=FourVector, type_to=CylindricalPos)  # type: ignore[arg-type]
-def convert_4vec_to_cylindrical(obj: FourVector, /) -> CylindricalPos:
+@conversion_method(type_from=FourVector, type_to=cxv.CylindricalPos)  # type: ignore[arg-type]
+def convert_4vec_to_cylindrical(obj: FourVector, /) -> cxv.CylindricalPos:
     """Convert a 4-vector to a Cylindrical 3-vector.
 
     Examples
@@ -112,11 +106,11 @@ def convert_4vec_to_cylindrical(obj: FourVector, /) -> CylindricalPos:
         [2.236 1.107 3.   ]>
 
     """
-    return convert(obj.q, CylindricalPos)
+    return convert(obj.q, cxv.CylindricalPos)
 
 
-@conversion_method(type_from=FourVector, type_to=SphericalPos)  # type: ignore[arg-type]
-def convert_4vec_to_spherical(obj: FourVector, /) -> SphericalPos:
+@conversion_method(type_from=FourVector, type_to=cxv.SphericalPos)  # type: ignore[arg-type]
+def convert_4vec_to_spherical(obj: FourVector, /) -> cxv.SphericalPos:
     """Convert a 4-vector to a spherical 3-vector.
 
     Examples
@@ -131,11 +125,11 @@ def convert_4vec_to_spherical(obj: FourVector, /) -> SphericalPos:
         [3.742 0.641 1.107]>
 
     """
-    return convert(obj.q, SphericalPos)
+    return convert(obj.q, cxv.SphericalPos)
 
 
-@conversion_method(type_from=FourVector, type_to=LonLatSphericalPos)  # type: ignore[arg-type]
-def convert_4vec_to_lonlat_spherical(obj: FourVector, /) -> LonLatSphericalPos:
+@conversion_method(type_from=FourVector, type_to=cxv.LonLatSphericalPos)  # type: ignore[arg-type]
+def convert_4vec_to_lonlat_spherical(obj: FourVector, /) -> cxv.LonLatSphericalPos:
     """Convert a 4-vector to a lon-lat spherical 3-vector.
 
     Examples
@@ -150,11 +144,11 @@ def convert_4vec_to_lonlat_spherical(obj: FourVector, /) -> LonLatSphericalPos:
         [ 1.107 53.301  3.742]>
 
     """
-    return convert(obj.q, LonLatSphericalPos)
+    return convert(obj.q, cxv.LonLatSphericalPos)
 
 
-@conversion_method(type_from=FourVector, type_to=MathSphericalPos)  # type: ignore[arg-type]
-def convert_4vec_to_mathsph(obj: FourVector, /) -> MathSphericalPos:
+@conversion_method(type_from=FourVector, type_to=cxv.MathSphericalPos)  # type: ignore[arg-type]
+def convert_4vec_to_mathsph(obj: FourVector, /) -> cxv.MathSphericalPos:
     """Convert a 4-vector to a math spherical 3-vector.
 
     Examples
@@ -169,4 +163,4 @@ def convert_4vec_to_mathsph(obj: FourVector, /) -> MathSphericalPos:
         [3.742 1.107 0.641]>
 
     """
-    return convert(obj.q, MathSphericalPos)
+    return convert(obj.q, cxv.MathSphericalPos)

…inax/_src/vectors/d4/register_convert.py …inax_astro/_src/vecs/register_convert.pysrc/coordinax/_src/vectors/d4/register_convert.py renamed to packages/coordinax-astro/src/coordinax_astro/_src/vecs/register_convert.py src/coordinax/_src/vectors/d4/register_convert.py renamed to packages/coordinax-astro/src/coordinax_astro/_src/vecs/register_convert.py

@@ -11,8 +11,8 @@
 import quaxed.numpy as jnp
 from dataclassish import replace
 
+import coordinax.vecs as cxv
 from .spacetime import FourVector
-from coordinax._src.vectors.d3 import AbstractPos3D
 
 
 @register(jax.lax.add_p)
@@ -32,7 +32,9 @@ def add_p_4vs(self: FourVector, other: FourVector, /) -> FourVector:
         [3 5 7 9]>
 
     """
-    return replace(self, t=self.t + other.t, q=cast(AbstractPos3D, self.q + other.q))
+    return replace(
+        self, t=self.t + other.t, q=cast(cxv.AbstractPos3D, self.q + other.q)
+    )
 
 
 @register(jax.lax.broadcast_in_dim_p)
@@ -44,7 +46,7 @@ def broadcast_in_dim_p_4v(
     return replace(
         obj,
         t=jnp.broadcast_to(obj.t, batch),
-        q=cast(AbstractPos3D, jnp.broadcast_to(obj.q, (*batch, 3))),  # type: ignore[arg-type]
+        q=cast(cxv.AbstractPos3D, jnp.broadcast_to(obj.q, (*batch, 3))),  # type: ignore[arg-type]
     )
 
 
@@ -83,7 +85,7 @@ def neg_p_4v(self: FourVector, /) -> FourVector:
         [-1 -1 -2 -3]>
 
     """
-    return replace(self, t=-self.t, q=cast(AbstractPos3D, -self.q))
+    return replace(self, t=-self.t, q=cast(cxv.AbstractPos3D, -self.q))
 
 
 @register(jax.lax.sub_p)
@@ -103,4 +105,4 @@ def sub_p_4v_4v(lhs: FourVector, rhs: FourVector, /) -> FourVector:
         [-1 -3 -3 -3]>
 
     """
-    return replace(lhs, t=lhs.t - rhs.t, q=cast(AbstractPos3D, lhs.q - rhs.q))
+    return replace(lhs, t=lhs.t - rhs.t, q=cast(cxv.AbstractPos3D, lhs.q - rhs.q))