Fix: Prevent numpy upgrading in CI during pip install

Author: elkins

.github/workflows/test.yml

@@ -62,10 +62,10 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install "numpy==${{ matrix.numpy-version }}"
-                if [ "${{ matrix.jax-version }}" != "latest" ]; then
-                  pip install "jax==${{ matrix.jax-version }}" "jaxlib==${{ matrix.jax-version }}"
-                fi
-        pip install -e ".[dev]"
+        if [ "${{ matrix.jax-version }}" != "latest" ]; then
+          pip install "jax==${{ matrix.jax-version }}" "jaxlib==${{ matrix.jax-version }}" "numpy==${{ matrix.numpy-version }}"
+        fi
+        pip install -e ".[dev]" "numpy==${{ matrix.numpy-version }}"
 
       shell: bash