use Pax image for TE testing

NVIDIA · Nov 13, 2023 · dbaba5b · dbaba5b
1 parent 12a57eb
commit dbaba5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
@@ -175,7 +175,7 @@ jobs:
     if: inputs.ARCHITECTURE == 'amd64'  # arm64 runners n/a
     uses: ./.github/workflows/_test_te.yaml
     with:
-      TE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_FINAL }}
+      TE_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAG_FINAL }}
     secrets: inherit
 
   test-t5x: