build: use xmmutablemap (#145)

Signed-off-by: nstarman <[email protected]>
GalacticDynamics · Aug 7, 2024 · c0e3993 · c0e3993
1 parent fd1f51f
commit c0e3993
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@
     "astropy",
     "dataclassish",
     "equinox",
-    "immutable_map_jax @ git+https://github.com/GalacticDynamics/immutable_map_jax.git",
+    "xmmutablemap",
     "jax",
     "jaxlib",
     "jaxtyping",

diff --git a/src/coordinax/_space.py b/src/coordinax/_space.py
@@ -12,9 +12,9 @@
 import equinox as eqx
 import jax
 from astropy.units import PhysicalType as Dimension, get_physical_type
-from immutable_map_jax import ImmutableMap
 from jax import Device
 from plum import dispatch
+from xmmutablemap import ImmutableMap
 
 import quaxed.numpy as jnp
 from unxt import Quantity