From 8318dee69f05c4e98b3d6bba52c7350ac03ea748 Mon Sep 17 00:00:00 2001 From: lars Date: Thu, 6 Jun 2024 15:44:56 +0200 Subject: [PATCH] set backend environment variable directly --- .github/workflows/tests.yml | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index a003d444c..d44c1079e 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -20,6 +20,7 @@ jobs: defaults: run: shell: bash -el {0} + env: ${{ matrix.backend }} steps: - name: Checkout code @@ -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: |