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:
3333 python -m pip install pytest
3434
3535 # 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
3737 python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
3838
3939 if [ "${{ matrix.numpy-version }}" == "dev" ]; then
4040 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
4142 elif [ "${{ matrix.numpy-version }}" == "1.22" ]; then
4243 python -m pip install 'numpy==1.22.*'
4344 elif [ "${{ matrix.numpy-version }}" == "1.26" ]; then
4445 python -m pip install 'numpy==1.26.*'
4546 else
4647 python -m pip install numpy
48+ python -m pip install dask[array] jax[cpu] sparse ndonnx
4749 fi
4850
4951 - name : Dump pip environment
You can’t perform that action at this time.
0 commit comments