From 6a9808c877f769774d7cbe0193a0d30a3adf53db Mon Sep 17 00:00:00 2001 From: Vincent Roulet Date: Tue, 18 Jun 2024 10:56:07 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 644444382 --- .github/workflows/tests.yml | 2 +- pyproject.toml | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 28adca005..15b63a19a 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -21,7 +21,7 @@ jobs: include: - python-version: "3.9" os: "ubuntu-latest" - jax-version: "0.4.27" # Keep this in sync with version in pyproject.toml + jax-version: "0.4.27" # Keep version in sync with pyproject.toml and copy.bara.sky! steps: - uses: "actions/checkout@v2" diff --git a/pyproject.toml b/pyproject.toml index 775f8903d..51d971b73 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,7 +32,8 @@ classifiers = [ dependencies = [ "absl-py>=0.7.1", "chex>=0.1.86", - "jax>=0.4.27", # Keep this in sync with version in .github/workflows/tests.yml + # Keep jax, jaxlib versions in sync with .github/workflows/tests.yml + "jax>=0.4.27", "jaxlib>=0.4.27", "numpy>=1.18.0", "etils[epy]",