File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -33,17 +33,19 @@ jobs:
33
33
python -m pip install pytest
34
34
35
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 ndonnx
36
+ python -m pip install array-api-strict
37
37
python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
38
38
39
39
if [ "${{ matrix.numpy-version }}" == "dev" ]; then
40
40
python -m pip install numpy --pre --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
41
+ python -m pip install dask[array] jax[cpu] sparse ndonnx
41
42
elif [ "${{ matrix.numpy-version }}" == "1.22" ]; then
42
43
python -m pip install 'numpy==1.22.*'
43
44
elif [ "${{ matrix.numpy-version }}" == "1.26" ]; then
44
45
python -m pip install 'numpy==1.26.*'
45
46
else
46
47
python -m pip install numpy
48
+ python -m pip install dask[array] jax[cpu] sparse ndonnx
47
49
fi
48
50
49
51
- name : Dump pip environment
You can’t perform that action at this time.
0 commit comments