File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -32,20 +32,22 @@ jobs:
32
32
python -m pip install --upgrade pip
33
33
python -m pip install pytest
34
34
35
+ # Don't `pip install .[dev]` as it would pull in the whole torch cuda stack
36
+ python -m pip install array-api-strict dask[array] jax[cpu] sparse
37
+ python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
38
+ if [ "${{ matrix.python-version }}" != "3.13" ]; then
39
+ # onnx wheels are not available on Python 3.13 at the moment of writing
40
+ python -m pip install ndonnx
41
+ fi
42
+
35
43
if [ "${{ matrix.numpy-version }}" == "dev" ]; then
36
44
python -m pip install numpy --pre --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
37
45
elif [ "${{ matrix.numpy-version }}" == "1.22" ]; then
38
46
python -m pip install 'numpy==1.22.*'
39
47
elif [ "${{ matrix.numpy-version }}" == "1.26" ]; then
40
48
python -m pip install 'numpy==1.26.*'
41
49
else
42
- # Don't `pip install .[dev]` as it would pull in the whole torch cuda stack
43
- python -m pip install array-api-strict dask[array] jax[cpu] numpy sparse
44
- python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
45
- if [ "${{ matrix.python-version }}" != "3.13" ]; then
46
- # onnx wheels are not available on Python 3.13 at the moment of writing
47
- python -m pip install ndonnx
48
- fi
50
+ python -m pip install numpy
49
51
fi
50
52
51
53
- name : Dump pip environment
You can’t perform that action at this time.
0 commit comments