Remove subhalo module and jax_cosmo dependency (#44)

* remove niche subhalo module and jax_cosmo dep * add explicit jaxlib dep
GalacticDynamics · Jan 2, 2024 · 9581b87 · 9581b87
1 parent 2826805
commit 9581b87
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 77 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,7 +34,7 @@ dependencies = [
   "diffrax",
   "equinox",
   "jax",
-  "jax_cosmo",
+  "jaxlib",
   "typing_extensions",
 ]
 

diff --git a/src/galax/potential/_potential/subhalo.py b/src/galax/potential/_potential/subhalo.py