wip

NVIDIA · Apr 30, 2024 · 033c25a · 033c25a
1 parent ddcec95
commit 033c25a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
@@ -289,7 +289,7 @@ jobs:
   # base, jax
   test-jax:
     needs: build-jax
-    if: inputs.ARCHITECTURE == 'amd64' # arm64 runners n/a
+    if: contains(fromJSON('["base", "jax"]'), inputs.TEST_SUBSET) && inputs.ARCHITECTURE == 'amd64' # arm64 runners n/a
     uses: ./.github/workflows/_test_unit.yaml
     with:
       TEST_NAME: jax