Skip to content

remove within-process test parallelization for tox to allow for bette… #631

remove within-process test parallelization for tox to allow for bette…

remove within-process test parallelization for tox to allow for bette… #631

Workflow file for this run

name: Tests
on:
pull_request:
push:
branches:
- master
- dev
- streamlined-backend
jobs:
test:
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-latest, windows-latest]
python-version: ["3.10", "3.11"]
backend: ["jax", "tensorflow", "torch"]
defaults:
run:
shell: bash -el {0}
env:
KERAS_BACKEND: ${{ matrix.backend }}
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Conda
uses: conda-incubator/setup-miniconda@v3
with:
environment-file: environment.yaml
python-version: ${{ matrix.python-version }}
- name: Install JAX
if: ${{ matrix.backend == 'jax' }}
run: |
pip install -U "jax[cpu]"
- name: Install NumPy
if: ${{ matrix.backend == 'numpy' }}
run: |
conda install numpy
- name: Install Tensorflow
if: ${{ matrix.backend == 'tensorflow' }}
run: |
pip install -U tensorflow
- name: Install PyTorch
if: ${{ matrix.backend == 'torch' }}
run: |
conda install pytorch torchvision torchaudio cpuonly -c pytorch
- name: Show Environment Info
run: |
conda info
conda list
conda config --show-sources
conda config --show
printenv | sort
- name: Run JAX Tests
if: ${{ matrix.backend == 'jax' }}
run: |
python -m pytest tests/ -v -m "not (numpy or tensorflow or torch)"
- name: Run NumPy Tests
if: ${{ matrix.backend == 'numpy' }}
run: |
python -m pytest tests/ -v -m "not (jax or tensorflow or torch)"
- name: Run TensorFlow Tests
if: ${{ matrix.backend == 'tensorflow' }}
run: |
python -m pytest tests/ -v -m "not (jax or numpy or torch)"
- name: Run PyTorch Tests
if: ${{ matrix.backend == 'torch' }}
run: |
python -m pytest tests/ -v -m "not (jax or numpy or tensorflow)"