Fix CI (#301)

SSAGESLabs · Feb 13, 2024 · 802cc16 · 802cc16
2 parents cd6b0a9 + dec0cf9
commit 802cc16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -32,7 +32,7 @@ jobs:
       - name: Install python dependecies
         run: |
           python -m pip install --upgrade pip
-          pip install ase dill "dm-haiku<0.0.11" "e3nn-jax!=0.20.4" jaxlib "jax-md>=0.2.7" jaxopt pytest matplotlib
+          pip install ase dill "dm-haiku<0.0.11" "e3nn-jax!=0.20.4" "jax<=0.4.15" "jaxlib<=0.4.15" "jax-md>=0.2.7" jaxopt pytest matplotlib
 
       - name: Install pysages
         run: pip install .
@@ -60,7 +60,7 @@ jobs:
       - name: Install python dependecies
         run: |
           python -m pip install --upgrade pip
-          pip install dill "dm-haiku<0.0.11" "e3nn-jax!=0.20.4" jaxlib "jax-md>=0.2.7" jaxopt pytest pylint flake8
+          pip install dill "dm-haiku<0.0.11" "e3nn-jax!=0.20.4" "jax<=0.4.15" "jaxlib<=0.4.15" "jax-md>=0.2.7" jaxopt pytest pylint flake8
           pip install -r docs/requirements.txt
       - name: Install pysages
         run: pip install .