From d47b6baffcebfd21e53aa2fa21d836169c802c80 Mon Sep 17 00:00:00 2001 From: Yu Li Date: Tue, 4 Aug 2026 23:08:13 +0000 Subject: [PATCH] Install test dependencies for unit testing --- .github/workflows/test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index e5060bf..a93baef 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -50,9 +50,9 @@ jobs: python-version: ${{ matrix.python-version }} - run: pip --version - if: matrix.jax-version == 'nightly' - run: pip install -e . -U --pre jax jaxlib -i https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/ + run: pip install -e ".[test]" -U --pre jax jaxlib -i https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/ - if: matrix.jax-version != 'nightly' - run: pip install -e . "jax==${JAX_VERSION}" + run: pip install -e ".[test]" "jax==${JAX_VERSION}" env: JAX_VERSION: ${{ matrix.jax-version }} - run: pip freeze