From ccafb52460b881a9ad072fd40e7dac887be81ff3 Mon Sep 17 00:00:00 2001
From: "Yu-Hang \"Maxin\" Tang" <Tang.Maxin@gmail.com>
Date: Mon, 13 Nov 2023 07:57:54 -0800
Subject: [PATCH] Fix job dependency

---
 .github/workflows/_ci.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
index 6f0cd8d8c..1447f5076 100644
--- a/.github/workflows/_ci.yaml
+++ b/.github/workflows/_ci.yaml
@@ -171,7 +171,7 @@ jobs:
     secrets: inherit
 
   test-te:
-    needs: build-jax
+    needs: build-pax
     if: inputs.ARCHITECTURE == 'amd64'  # arm64 runners n/a
     uses: ./.github/workflows/_test_te.yaml
     with:
@@ -200,4 +200,4 @@ jobs:
   #   with:
   #     ROSETTA_T5X_IMAGE: ${{ needs.build-rosetta-t5x.outputs.DOCKER_TAGS }}
   #   secrets: inherit 
- 
\ No newline at end of file
+