⬆️ dep-bump: unxt, coordinax (#784)

Author: nstarman

pyproject.toml

@@ -29,7 +29,7 @@
   dependencies = [
     "astropy>=7.0",
     "beartype==0.21.0", # TODO: figure out why 0.22.0 breaks things
-    "coordinax>=0.23.1",
+    "coordinax>=0.23.2",
     "dataclassish>=0.8.0",
     "diffrax>=0.7",
     "diffraxtra>=1.5.2",
@@ -51,7 +51,7 @@
     "quaxed>=0.10.2",
     "tfp-nightly[jax]>=0.25.0",
     "typing-extensions>=4.13.2",
-    "unxt>=1.7.7",
+    "unxt>=1.10.3",
     "wadler_lindig>=0.1.7",
     "xmmutablemap>=0.2.1",
     "zeroth>=1.0",

src/galax/_interop/galax_interop_astropy/potential.py

@@ -47,7 +47,7 @@ def parse_to_xyz_t(
 # =============================================================================
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload,misc]
+@dispatch(precedence=1)
 def potential(
     pot: gp.AbstractPotential,
     xyz: Real[APYQuantity, "*#batch 3"],
@@ -63,7 +63,7 @@ def potential(
     return gp.potential(pot, convert(xyz, FastQ), convert(t, FastQ))
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload,misc]
+@dispatch(precedence=1)
 def gradient(
     pot: gp.AbstractPotential,
     xyz: Real[APYQuantity, "*#batch 3"],
@@ -79,7 +79,7 @@ def gradient(
     return gp.gradient(pot, convert(xyz, FastQ), convert(t, FastQ))
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload,misc]
+@dispatch(precedence=1)
 def density(
     pot: gp.AbstractPotential,
     xyz: Real[APYQuantity, "*#batch 3"],
@@ -95,7 +95,7 @@ def density(
     return gp.density(pot, convert(xyz, FastQ), convert(t, FastQ))
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload,misc]
+@dispatch(precedence=1)
 def hessian(
     pot: gp.AbstractPotential,
     xyz: Real[APYQuantity, "*#batch 3"],

src/galax/coordinates/_src/base.py

@@ -279,9 +279,7 @@ def __getitem__(
     def __pdoc__(self, **kwargs: object) -> wl.AbstractDoc:
         return wl.bracketed(
             begin=wl.TextDoc(f"{self.__class__.__name__}("),
-            docs=wl.named_objs(
-                field_items(self), short_arrays=False, compact_arrays=True
-            ),
+            docs=wl.named_objs(field_items(self), short_arrays="compact"),
             sep=wl.comma,
             end=wl.TextDoc(")"),
             indent=kwargs.get("indent", 4),

src/galax/coordinates/_src/pscs/base_composite.py

@@ -255,7 +255,7 @@ def getitem(self: AbstractCompositePhaseSpaceCoordinate, key: str, /) -> Any:
 # `unxt.uconvert`
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload,misc]  # TODO: make precedence=0
+@dispatch(precedence=1)  # TODO: make precedence=0
 def uconvert(
     usys: u.AbstractUnitSystem | str, cwt: AbstractCompositePhaseSpaceCoordinate, /
 ) -> AbstractCompositePhaseSpaceCoordinate:

src/galax/coordinates/_src/pscs/base_single.py

@@ -63,7 +63,7 @@ class AbstractBasicPhaseSpaceCoordinate(AbstractPhaseSpaceCoordinate):
 # `unxt.uconvert` dispatches
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload, misc]  # TODO: make precedence=0
+@dispatch(precedence=1)  # TODO: make precedence=0
 def uconvert(
     units: u.AbstractUnitSystem | str, wt: AbstractBasicPhaseSpaceCoordinate
 ) -> AbstractBasicPhaseSpaceCoordinate:

src/galax/coordinates/_src/psps/core.py

@@ -309,7 +309,7 @@ def vconvert(
 # `unxt.uconvert` dispatches
 
 
-@dispatch(precedence=1)  # type: ignore[call-overload, misc]  # TODO: make precedence=0
+@dispatch(precedence=1)  # TODO: make precedence=0
 def uconvert(
     units: u.AbstractUnitSystem | str, psp: PhaseSpacePosition
 ) -> PhaseSpacePosition:

src/galax/dynamics/_src/parsetime.py

@@ -1,6 +1,6 @@
 """Utilities for dynamics."""
 
-__all__ = ["parse_time_specification"]
+__all__ = ("parse_time_specification",)
 
 
 from collections.abc import Sequence
@@ -12,7 +12,6 @@
 
 import quaxed.numpy as jnp
 import unxt as u
-from unxt._src.units.api import AstropyUnits as Unit
 
 
 def parse_time_specification(
@@ -163,7 +162,7 @@ def parse_time_specification(
 # -----------------------------------------------
 
 
-def _parse_to_time_unit(obj: Any, /) -> Unit | None:
+def _parse_to_time_unit(obj: Any, /) -> u.AbstractUnit | None:
     return u.unitsystem(obj)["time"] if obj is not None else None
 
 

src/galax/potential/_src/params/base.py

@@ -1,16 +1,13 @@
 """Parameters on a Potential."""
 
-__all__ = [
-    "ParameterCallable",
-    "AbstractParameter",
-]
+__all__ = ("ParameterCallable", "AbstractParameter")
 
 import abc
 from typing import Any, Protocol, runtime_checkable
 
 import equinox as eqx
 
-from unxt._src.units.api import AstropyUnits
+import unxt as u
 
 import galax._custom_types as gt
 
@@ -20,7 +17,7 @@ class ParameterCallable(Protocol):
     """Protocol for a Parameter callable."""
 
     def __call__(
-        self, t: gt.BBtQuSz0, *, ustrip: AstropyUnits | None = None, **kwargs: Any
+        self, t: gt.BBtQuSz0, *, ustrip: u.AbstractUnit | None = None, **kwargs: Any
     ) -> gt.QuSzAny | gt.SzAny:
         """Compute the parameter value at the given time(s).
 
@@ -56,7 +53,7 @@ class AbstractParameter(eqx.Module):  # type: ignore[misc]
 
     @abc.abstractmethod
     def __call__(
-        self, t: gt.BBtQuSz0, *, ustrip: AstropyUnits | None = None, **kwargs: Any
+        self, t: gt.BBtQuSz0, *, ustrip: u.AbstractUnit | None = None, **kwargs: Any
     ) -> gt.QuSzAny:
         """Compute the parameter value at the given time(s).
 

src/galax/potential/_src/params/constant.py

@@ -16,7 +16,6 @@
 
 import unxt as u
 from dataclassish.converters import Unless
-from unxt._src.units.api import AstropyUnits
 from unxt.quantity import AllowValue
 
 import galax._custom_types as gt
@@ -128,7 +127,7 @@ def __call__(
         self,
         t: gt.BBtQuSz0 = t0,  # noqa: ARG002
         *,
-        ustrip: AstropyUnits | None = None,
+        ustrip: u.AbstractUnit | None = None,
         **__: Any,
     ) -> gt.QuSzAny:
         """Return the constant parameter value.

src/galax/potential/_src/params/core.py

@@ -13,7 +13,6 @@
 import jax.core
 
 import unxt as u
-from unxt._src.units.api import AstropyUnits
 from unxt.quantity import AllowValue
 
 import galax._custom_types as gt
@@ -84,7 +83,7 @@ def __check_init__(self) -> None:
 
     @ft.partial(jax.jit, static_argnames=("ustrip",))
     def __call__(
-        self, t: gt.BBtQuSz0, *, ustrip: AstropyUnits | None = None, **_: Any
+        self, t: gt.BBtQuSz0, *, ustrip: u.AbstractUnit | None = None, **_: Any
     ) -> gt.QuSzAny | gt.SzAny:
         """Return the parameter value.
 
@@ -150,7 +149,7 @@ class CustomParameter(AbstractParameter):
 
     @ft.partial(jax.jit, static_argnames=("ustrip",))
     def __call__(
-        self, t: gt.BBtQuSz0, *, ustrip: AstropyUnits | None = None, **kwargs: Any
+        self, t: gt.BBtQuSz0, *, ustrip: u.AbstractUnit | None = None, **kwargs: Any
     ) -> gt.QuSzAny | gt.SzAny:
         out = self.func(t, **kwargs)
         return out if ustrip is None else u.ustrip(AllowValue, ustrip, out)