Pin jaxlib to a version <= 0.1.75

This prevents low-level errors recently introduced by newer versions of `jaxlib`.
aesara-devs · Feb 19, 2022 · 9ef57ac · 9ef57ac
1 parent 9b4531d
commit 9ef57ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -143,7 +143,7 @@ jobs:
         run: |
           mamba install --yes -q "python~=${PYTHON_VERSION}=*_cpython" mkl numpy scipy pip mkl-service graphviz cython pytest coverage pytest-cov sympy
           if [[ $INSTALL_NUMBA == "1" ]]; then mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" "numba>=0.55" numba-scipy; fi
-          mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" jax jaxlib
+          mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" jax "jaxlib <= 0.1.75"
           pip install -e ./
           mamba list && pip freeze
           python -c 'import aesara; print(aesara.config.__str__(print_doc=False))'

diff --git a/requirements.txt b/requirements.txt
@@ -11,7 +11,7 @@ cython
 sympy
 versioneer
 jax
-jaxlib
+jaxlib<=0.1.75
 numba>=0.55
 numba-scipy>=0.3.0
 diff-cover