downgrade jax for spatial

DendrouLab · Mar 11, 2024 · 5687231 · 5687231
1 parent 95c5c48
commit 5687231
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,8 +20,8 @@ classifiers = [
 requires-python = ">= 3.10"
 
 dependencies = [
-    "jax==0.4.23",
-    "jaxlib==0.4.23",
+    "jax",
+    "jaxlib",
     "apsw",
     "bbknn",
     "cgatcore",
@@ -59,6 +59,8 @@ notebook = [
     "jupyterlab"
 ]
 spatial = [
+    "jax==0.4.23",
+    "jaxlib==0.4.23",
     "squidpy",
     "cell2location",
     "tangram-sc"