Skip to content

Commit

Permalink
Fix env variable expressions.
Browse files Browse the repository at this point in the history
  • Loading branch information
alxmrs committed Jul 23, 2024
1 parent f68194f commit 6b6dc4f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/jax-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,6 @@ jobs:
pytest -k "not argmax and not argmin and not mean and not apply_reduction and not broadcast_trick and not groupby and not random and not visualization and not plan_scaling and not optimization"
env:
CUBED_BACKEND_ARRAY_API_MODULE: jax.numpy
JAX_ENABLE_X64: matrix.os != 'macos-14'
CUBED_DEFAULT_PRECISION_X32: matrix.os == 'macos-14'
JAX_ENABLE_X64: ${{ matrix.os != 'macos-14' }}
CUBED_DEFAULT_PRECISION_X32: ${{ matrix.os == 'macos-14' }}
ENABLE_PJRT_COMPATIBILITY: True

0 comments on commit 6b6dc4f

Please sign in to comment.