Skip to content

Commit c69accf

Browse files
committed
Tweak CI job split to balance runtime
1 parent acae31e commit c69accf

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

.github/workflows/test.yml

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -84,14 +84,12 @@ jobs:
8484
install-torch: [0]
8585
install-xarray: [0]
8686
part:
87-
- "tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/sparse --ignore=tests/xtensor"
87+
- "tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/xtensor"
8888
- "tests/scan"
89-
- "tests/sparse"
90-
- "tests/tensor --ignore=tests/tensor/conv --ignore=tests/tensor/rewriting --ignore=tests/tensor/test_math.py --ignore=tests/tensor/test_basic.py --ignore=tests/tensor/test_inplace.py --ignore=tests/tensor/test_blas.py --ignore=tests/tensor/test_elemwise.py --ignore=tests/tensor/test_math_scipy.py"
91-
- "tests/tensor/conv"
89+
- "tests/tensor --ignore=tests/tensor/rewriting --ignore=tests/tensor/test_math.py --ignore=tests/tensor/test_basic.py --ignore=tests/tensor/test_inplace.py --ignore=tests/tensor/conv --ignore=tests/tensor/test_blas.py --ignore=tests/tensor/test_elemwise.py --ignore=tests/tensor/test_math_scipy.py"
9290
- "tests/tensor/rewriting"
9391
- "tests/tensor/test_math.py"
94-
- "tests/tensor/test_basic.py tests/tensor/test_inplace.py"
92+
- "tests/tensor/test_basic.py tests/tensor/test_inplace.py tests/tensor/conv"
9593
- "tests/tensor/test_blas.py tests/tensor/test_elemwise.py tests/tensor/test_math_scipy.py"
9694
exclude:
9795
- python-version: "3.10"
@@ -130,7 +128,9 @@ jobs:
130128
numpy-version: "~=2.1.0"
131129
fast-compile: 0
132130
float32: 0
133-
part: "tests/link/numba"
131+
part:
132+
- "tests/link/numba --ignore=tests/link/test_slinalg.py"
133+
- "tests/link/numba/test_slinalg.py"
134134
- install-jax: 1
135135
os: "ubuntu-latest"
136136
python-version: "3.10"
@@ -207,7 +207,7 @@ jobs:
207207
if [[ $INSTALL_TORCH == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" pytorch pytorch-cuda=12.1 "mkl<=2024.0" -c pytorch -c nvidia; fi
208208
if [[ $INSTALL_XARRAY == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" xarray xarray-einstats; fi
209209
pip install pytest-sphinx
210-
210+
211211
pip install -e ./
212212
micromamba list && pip freeze
213213
python -c 'import pytensor; print(pytensor.config.__str__(print_doc=False))'
@@ -218,7 +218,7 @@ jobs:
218218
fi
219219
env:
220220
PYTHON_VERSION: ${{ matrix.python-version }}
221-
NUMPY_VERSION: ${{ matrix.numpy-version }}
221+
NUMPY_VERSION: ${{ matrix.numpy-version }}
222222
INSTALL_NUMBA: ${{ matrix.install-numba }}
223223
INSTALL_JAX: ${{ matrix.install-jax }}
224224
INSTALL_TORCH: ${{ matrix.install-torch}}
@@ -343,4 +343,3 @@ jobs:
343343
directory: ./coverage/
344344
fail_ci_if_error: true
345345
token: ${{ secrets.CODECOV_TOKEN }}
346-

0 commit comments

Comments
 (0)