Tweak build.yml again

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Jun 14, 2024 · df3b907 · df3b907
1 parent 8c159cc
commit df3b907
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -50,8 +50,9 @@ jobs:
       run: pdm run pytest -v --shorter-than=1.0 --cov=project --cov-report=xml --cov-append
     - name: Test with pytest (fast)
       env:
+        # todo: why isn't this sufficient?
         JAX_PLATFORMS: cpu
-      run: pdm run pytest -v --cov=project --cov-report=xml --cov-append
+      run: JAX_PLATFORMS=cpu pdm run pytest -v --cov=project --cov-report=xml --cov-append
 
     - name: Store coverage report as an artifact
       uses: actions/upload-artifact@v4