mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pin numpy~=2.1.0 in workflow file instead of test-requirements.txt

Browse Source 
PiperOrigin-RevId: 737632771
This commit is contained in:
Nitin Srinivasan 2025-03-17 08:58:18 -07:00 committed by jax authors
parent 3649da56fc
commit 031614c22b
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/pytest_cpu.yml vendored
View File

@ -118,6 +118,11 @@ jobs:
run: | run: |
$JAXCI_PYTHON -m pip install uv~=0.5.30 $JAXCI_PYTHON -m pip install uv~=0.5.30
$JAXCI_PYTHON -m uv pip install -r build/test-requirements.txt $JAXCI_PYTHON -m uv pip install -r build/test-requirements.txt
# CPU Pytests crash with NumPy 2.2+ on Linux Aarch64; b/399168632
if [[ $OS == "linux" && $ARCH == "aarch64" ]]; then
$JAXCI_PYTHON -m uv pip install numpy~=2.1.0
fi
# Halt for testing # Halt for testing
- name: Wait For Connection - name: Wait For Connection
uses: google-ml-infra/actions/ci_connection@main uses: google-ml-infra/actions/ci_connection@main

5
build/test-requirements.txt
View File

@ -18,7 +18,4 @@ setuptools
matplotlib~=3.8.4; python_version=="3.10" matplotlib~=3.8.4; python_version=="3.10"
matplotlib; python_version>="3.11" matplotlib; python_version>="3.11"
opt-einsum opt-einsum
auditwheel auditwheel
# CPU Pytests crash with NumPy 2.2+ on Linux Aarch64; b/399168632
numpy~=2.1.0; platform_system == "Linux" and platform_machine == "aarch64"