Use older version of JAX to work with Ubuntu numpy

Author: benmwebb

.github/workflows/ci.yml

@@ -24,6 +24,8 @@ jobs:
           linkflags: "--coverage"
           cgaldir: "/usr/lib/x86_64-linux-gnu/cmake/CGAL/"
           skiptests: "expensive|medium|IMP\\.modeller\\-"
+          # Newer versions of jax need numpy 2 (Ubuntu 24.04 has numpy 1)
+          pip_extra_packages: jax==0.7.1
           show_build: Debug
           cmake_build: Debug
           max_checks: INTERNAL
@@ -60,7 +62,7 @@ jobs:
             libprotobuf-dev protobuf-compiler \
             libopenmpi-dev python3-dev python3-numpy python3-protobuf \
             python3-nose python3-pip python3-biopython
-          pip3 install codecov
+          pip3 install codecov ${{ matrix.pip_extra_packages }}
       - name: Set up Python ${{ matrix.python-version }}
         if: matrix.os == 'macos-latest'
         uses: actions/setup-python@v6