NVIDIA / JAX-Toolbox

JAX-Toolbox
Apache License 2.0
210 stars 35 forks source link

Ping numpy to 1.24.3 #908

Closed DwarKapex closed 3 weeks ago

DwarKapex commented 4 weeks ago

Jax-Core container fails to build with Recently released numpy 2. Pin numpy to 1.26.2 until TF adds support for numpy 2.

olupton commented 4 weeks ago

Looks like some of the test failures are still related to numpy 2: https://github.com/NVIDIA/JAX-Toolbox/actions/runs/9573423195/job/26395865704#step:4:84

nouiz commented 3 weeks ago

I'll merge now to have at least a partial fix. Hopefully, that way tomorrow CI will be better.