diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index 1d35611537..9b81ab7d82 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -83,7 +83,7 @@ jobs: package-name: pylibraft test-before-amd64: "pip install cupy-cuda11x" # On arm also need to install cupy from the specific webpage. - test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64" + test-before-arm64: "pip install 'cupy-cuda11x<12.0.0' -f https://pip.cupy.dev/aarch64" test-unittest: "python -m pytest -v ./python/pylibraft/pylibraft/test" test-smoketest: "python ./ci/wheel_smoke_test_pylibraft.py" wheel-build-raft-dask: diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index aa7ca21b5f..876c9a6722 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -40,7 +40,7 @@ jobs: sha: ${{ inputs.sha }} package-name: pylibraft test-before-amd64: "pip install cupy-cuda11x" - test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64" + test-before-arm64: "pip install 'cupy-cuda11x<12.0.0' -f https://pip.cupy.dev/aarch64" test-unittest: "python -m pytest -v ./python/pylibraft/pylibraft/test" wheel-tests-raft-dask: secrets: inherit