Support most recent unxt (#484)

* ⬆️ dep-bump: unxt v1.10.3+
* 🚨 fix-lint: mypy
* 🚨 fix-lint: mypy

Signed-off-by: nstarman <[email protected]>
(cherry picked from commit 8ab58a3)
Signed-off-by: nstarman <[email protected]>

Author: nstarman

pyproject.toml

@@ -32,7 +32,7 @@ dependencies = [
   "quax-blocks>=0.4.0",
   "quaxed>=0.10.4",
   "typing-extensions>=4.13.2",
-  "unxt>=1.7.3",
+  "unxt>=1.10.3",
   "wadler-lindig>=0.1.6",
   "xmmutablemap>=0.1",
 ]

src/coordinax/_coordinax_space_frames/frame_transforms.py

@@ -33,7 +33,7 @@
 # ---------------------------------------------------------------
 # Base Space-Frame Transformation
 
-_icrs_frame = ICRS()  # type: ignore[no-untyped-call]
+_icrs_frame = ICRS()
 
 
 @dispatch

src/coordinax/_src/distances/measures.py

@@ -12,7 +12,6 @@
 
 import quaxed.numpy as jnp
 import unxt as u
-from unxt._src.units.api import AstropyUnits
 
 from .base import AbstractDistance
 
@@ -46,7 +45,7 @@ class Distance(AbstractDistance):
     )
     """The value of the `unxt.AbstractQuantity`."""
 
-    unit: AstropyUnits = eqx.field(static=True, converter=u.unit)
+    unit: u.AbstractUnit = eqx.field(static=True, converter=u.unit)
     """The unit associated with this value."""
 
     _: KW_ONLY
@@ -103,7 +102,7 @@ class DistanceModulus(AbstractDistance):
     )
     """The value of the `unxt.AbstractQuantity`."""
 
-    unit: AstropyUnits = eqx.field(static=True, converter=u.unit)
+    unit: u.AbstractUnit = eqx.field(static=True, converter=u.unit)
     """The unit associated with this value."""
 
     def __check_init__(self) -> None:
@@ -138,7 +137,7 @@ class Parallax(AbstractDistance):
     To disable this check, set `check_negative=False`.
 
     >>> Parallax(-1, "mas", check_negative=False)
-    Parallax(Array(-1, dtype=int32, ...), unit='mas', check_negative=False)
+    Parallax(Array(-1, dtype=int32, weak_type=True), unit='mas')
 
     """
 
@@ -147,7 +146,7 @@ class Parallax(AbstractDistance):
     )
     """The value of the `unxt.AbstractQuantity`."""
 
-    unit: AstropyUnits = eqx.field(static=True, converter=u.unit)
+    unit: u.AbstractUnit = eqx.field(static=True, converter=u.unit)
     """The unit associated with this value."""
 
     _: KW_ONLY

src/coordinax/_src/operators/base.py

@@ -123,7 +123,7 @@ def __pdoc__(self, **kwargs: Any) -> wl.AbstractDoc:
             if not jnp.all(v == getattr(self.__class__, k, _sentinel))
         ]
         # Make the field docs list
-        if len(fitems) == 1 and kwargs.get("compact_arrays", False):
+        if len(fitems) == 1 and kwargs.get("short_arrays") == "compact":
             docs = [wl.TextDoc(str(fitems[0][1]))]
         else:
             docs = wl.named_objs(fitems, **kwargs)
@@ -173,9 +173,7 @@ def __str__(self) -> str:
         )
 
         """
-        return wl.pformat(
-            self, width=88, vector_form=True, short_arrays=False, compact_arrays=True
-        )
+        return wl.pformat(self, width=88, vector_form=True, short_arrays="compact")
 
     # ===========================================
     # Operator Composition

src/coordinax/_src/operators/galilean/rotation.py

@@ -446,7 +446,7 @@ def simplify_op(op: GalileanRotation, /, **kwargs: Any) -> AbstractOperator:
 
     """
     if jnp.allclose(op.rotation, jnp.eye(3), **kwargs):
-        return Identity()  # type: ignore[no-untyped-call]
+        return Identity()
     return op
 
 

src/coordinax/_src/vectors/base/base.py

@@ -737,9 +737,7 @@ def __repr__(self) -> str:
         representation of the vector.
 
         """
-        return wl.pformat(
-            self, vector_form=False, short_arrays=False, compact_arrays=True
-        )
+        return wl.pformat(self, vector_form=False, short_arrays="compact")
 
     def __str__(self) -> str:
         """Return a string representation of the vector-like object."""

src/coordinax/_src/vectors/base/vector.py

@@ -562,7 +562,7 @@ def __pdoc__(self, *, vector_form: bool = False, **kwargs: Any) -> wl.AbstractDo
                     begin=wl.TextDoc(" "),  # indent to opening "<"
                     docs=wl.named_objs(
                         [(k, getattr(self, k)) for k in self._AUX_FIELDS],
-                        **kwargs | {"short_arrays": False, "compact_arrays": True},
+                        **kwargs | {"short_arrays": "compact"},
                     ),
                     sep=wl.comma,
                     end=wl.TextDoc(""),  # no end bracket

src/coordinax/_src/vectors/collection/utils.py

@@ -9,7 +9,7 @@
 
 import quaxed.numpy as jnp
 import unxt as u
-from unxt._src.dimensions.core import name_of
+from unxt._src.dimensions import name_of
 
 DimensionLike: TypeAlias = Dimension | str
 Shape: TypeAlias = tuple[int, ...]

src/coordinax/_src/vectors/converters.py

@@ -8,10 +8,7 @@
 _2pid = u.Angle(360, "deg")
 
 
-def converter_azimuth_to_range(
-    phi: u.AbstractQuantity,
-    /,
-) -> u.AbstractQuantity:
+def converter_azimuth_to_range(phi: u.AbstractQuantity, /) -> u.AbstractQuantity:
     """Wrap a polar angle to the range [0, 2pi).
 
     It's safe to do this conversion since this is a phase cut, unlike `theta`,

src/coordinax/_src/vectors/d4/spacetime.py

@@ -206,8 +206,8 @@ def __pdoc__(self, *, vector_form: bool = False, **kwargs: Any) -> wl.AbstractDo
             + wl.TextDoc(")")
         ).group()
 
-        vs = np.array2string(  # type: ignore[call-overload]  # TODO: use other method
-            jnp.stack(
+        vs = np.array2string(
+            jnp.stack(  # type: ignore[arg-type]
                 tuple(
                     u.ustrip(v)
                     for v in cast(