fix jaxlib config name

google · May 6, 2024 · aac3679 · aac3679
1 parent 28fe45d
commit aac3679
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/metal_plugin_ci.yml b/.github/workflows/metal_plugin_ci.yml
@@ -17,7 +17,7 @@ jobs:
     strategy:
       fail-fast: false # don't cancel all jobs on failure
       matrix:
-        jaxlib-version: ["plugin_latest", "nightly"]
+        jaxlib-version: ["pypi_latest", "nightly"]
     name: "Jax-Metal plugin test (jaxlib=${{ matrix.jaxlib-version }})"
     runs-on: [self-hosted, macOS, ARM64]