Fix CI

SSAGESLabs · Feb 13, 2024 · 84d9391 · 84d9391
1 parent cd6b0a9
commit 84d9391
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" "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" "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 .