moved jax back to main dependencies

testing python 3.11
DendrouLab · Feb 12, 2024 · c9357b1 · c9357b1
1 parent e38967d
commit c9357b1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/.github/workflows/ingestion-ci.yml b/.github/workflows/ingestion-ci.yml
@@ -18,7 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["ubuntu-latest"] # , "macos-latest", "windows-latest"
-        python-version: ["3.9"]
+        python-version: ["3.11"]
 
     steps:
       - uses: actions/checkout@v4

diff --git a/.github/workflows/spatial-ci.yml b/.github/workflows/spatial-ci.yml
@@ -18,7 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ["ubuntu-latest"] # , "macos-latest", "windows-latest"
-        python-version: ["3.9"]
+        python-version: ["3.11"]
 
     steps:
       - uses: actions/checkout@v4

diff --git a/pyproject.toml b/pyproject.toml
@@ -36,7 +36,8 @@ dependencies = [
     "drmaa", 
     "gevent",
     "harmonypy",
-    #removed jax to figure if it's needed by cell2location
+    "jax<=0.4.23", # temp for scvi https://github.com/theislab/scvelo/issues/1186
+    "jaxlib<=0.4.23",
     "leidenalg", 
     "louvain",
     "matplotlib<=3.7.3",
@@ -72,9 +73,7 @@ notebook = [
 spatial = [
     "squidpy",
     "cell2location",
-    "tangram-sc",
-    "jax<=0.4.23",
-    "jaxlib<=0.4.23",
+    "tangram-sc"
 ]
 pypi_upload = [
     "twine",