diff --git a/.github/workflows/jax-tests.yml b/.github/workflows/jax-tests.yml index 01de4044..43b9229a 100644 --- a/.github/workflows/jax-tests.yml +++ b/.github/workflows/jax-tests.yml @@ -44,7 +44,7 @@ jobs: - name: Run tests run: | # exclude tests that rely on structured types since JAX doesn't support these - pytest -k "not argmax and not argmin and not mean and not apply_reduction and not broadcast_trick and not groupby and not object_dtype" + pytest -k "not argmax and not argmin and not mean and not std and not var and not apply_reduction and not broadcast_trick and not groupby and not object_dtype" env: CUBED_BACKEND_ARRAY_API_MODULE: jax.numpy JAX_ENABLE_X64: True