update test jax version

This commit is contained in:
Wassim Kabalan 2025-06-08 10:56:34 +02:00
parent 49c93aacf6
commit 6aacd81bd6

View file

@ -29,7 +29,7 @@ jobs:
run: | run: |
sudo apt-get install -y libopenmpi-dev sudo apt-get install -y libopenmpi-dev
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install jax==0.4.35 pip install jax
pip install numpy setuptools cython wheel pip install numpy setuptools cython wheel
pip install git+https://github.com/MP-Gadget/pfft-python pip install git+https://github.com/MP-Gadget/pfft-python
pip install git+https://github.com/MP-Gadget/pmesh pip install git+https://github.com/MP-Gadget/pmesh