Skip to content

Commit

Permalink
set backend environment variable directly
Browse files Browse the repository at this point in the history
  • Loading branch information
LarsKue committed Jun 6, 2024
1 parent 2653090 commit 8318dee
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ jobs:
defaults:
run:
shell: bash -el {0}
env: ${{ matrix.backend }}

steps:
- name: Checkout code
Expand All @@ -35,22 +36,18 @@ jobs:
if: ${{ matrix.backend == 'jax' }}
run: |
pip install -U "jax[cpu]"
conda env config vars set KERAS_BACKEND=jax
- name: Install NumPy
if: ${{ matrix.backend == 'numpy' }}
run: |
conda install numpy
conda env config vars set KERAS_BACKEND=numpy
- name: Install Tensorflow
if: ${{ matrix.backend == 'tensorflow' }}
run: |
pip install -U tensorflow
conda env config vars set KERAS_BACKEND=tensorflow
- name: Install PyTorch
if: ${{ matrix.backend == 'torch' }}
run: |
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda env config vars set KERAS_BACKEND=torch
- name: Show Environment Info
run: |
Expand Down

0 comments on commit 8318dee

Please sign in to comment.