refactor: use plum type_unparametric

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

src/unxt/_quantity/__init__.py

@@ -3,14 +3,13 @@
 Copyright (c) 2023 Galactic Dynamics. All rights reserved.
 """
 
-from . import base, base_parametric, core, distance, fast, functional, utils
+from . import base, base_parametric, core, distance, fast, functional
 from .base import *
 from .base_parametric import *
 from .core import *
 from .distance import *
 from .fast import *
 from .functional import *
-from .utils import *
 
 # isort: split
 # Register dispatches
@@ -23,4 +22,3 @@
 __all__ += distance.__all__
 __all__ += fast.__all__
 __all__ += functional.__all__
-__all__ += utils.__all__

src/unxt/_quantity/register_dispatches.py

@@ -7,13 +7,13 @@
 import jax.experimental.array_api as jax_xp
 import numpy as np
 from jax import Device
+from plum.parametric import type_unparametrized as type_np
 
 from quaxed._types import DType
 from quaxed.array_api._dispatch import dispatcher as dispatcher_
 
 from .base import AbstractQuantity
 from .core import Quantity
-from .utils import type_unparametrized as type_np
 
 T = TypeVar("T")
 

src/unxt/_quantity/register_primitives.py

@@ -21,13 +21,13 @@
 from jax.core import Primitive
 from jaxtyping import Array, ArrayLike
 from plum import promote
+from plum.parametric import type_unparametrized as type_np
 from quax import register as register_
 
 from .base import AbstractQuantity, can_convert_unit
 from .base_parametric import AbstractParametricQuantity
 from .core import Quantity
 from .distance import AbstractDistance
-from .utils import type_unparametrized as type_np
 from unxt._units import Unit
 
 T = TypeVar("T")

src/unxt/experimental.py

@@ -16,9 +16,9 @@
 import equinox as eqx
 import jax
 from jaxtyping import ArrayLike
+from plum.parametric import type_unparametrized
 
 from ._quantity.core import Quantity
-from ._quantity.utils import type_unparametrized
 from ._typing import Unit
 
 P = ParamSpec("P")