Improve chex test coverage

PiperOrigin-RevId: 715578386
google-deepmind · Jan 15, 2025 · 599ae5c · 599ae5c
1 parent fc6e9bf
commit 599ae5c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -23,14 +23,14 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
         os: [ubuntu-latest]
         jax-version: ["newest"]
         include:
           - python-version: "3.9"
             os: "ubuntu-latest"
             jax-version: "0.4.27"  # Keep this in sync with version in requirements.txt
-          - python-version: "3.11"
+          - python-version: "3.13"
             os: "ubuntu-latest"
             jax-version: "nightly"