Make AddLabel workflow in chain of "MaxText Package Tests".

 Make AddLabel workflow associated with PR, by adding on: pull_request
 Remove single commit enforcement, as it is not in practice
 Make AddLabel workflow exit(1) to show failed if it fails to add label.

Author: Charles Li

.github/workflows/AddLabel.yml

@@ -15,23 +15,37 @@
 name: Add Label
 
 on:
+  pull_request:  # include this workflow in PR triggered Actions
   workflow_run:
-    workflows: [Tests, CodeQL]
+    workflows: ["MaxText Package Tests"]
     types:
       - completed
   pull_request_review:
   pull_request_review_comment:
   workflow_dispatch:
 
 jobs:
+  WaitForCI:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Wait for CI workflow to succeed
+        uses: lewagon/wait-on-check-action@v1.3.1
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+          check-name: 'MaxText Package Tests' # The name of the job in Workflow A
+          repo-token: ${{ secrets.GITHUB_TOKEN }}
+          wait-interval: 10
+
   AddPullReady:
+    needs: [WaitForCI]
     permissions:
       checks: read
       pull-requests: write
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/github-script@v7
+      - name: Add Pull Request Label
+        uses: actions/github-script@v7
         with:
           script: |
             const owner = "google"
@@ -40,9 +54,10 @@ jobs:
             if (context.payload.pull_request !== undefined) {
               pull_number = context.payload.pull_request.number
             } else if (context.payload.workflow_run !== undefined) {
+              workflow_name = context.payload.workflow_run.name
               if (context.payload.workflow_run.pull_requests.length === 0) {
                 console.log("This workflow is NOT running within a PR's context")
-                process.exit()
+                process.exit(1)
               }
               console.log(context.payload.workflow_run.pull_requests)
               pull_number = context.payload.workflow_run.pull_requests[0].number
@@ -68,15 +83,15 @@ jobs:
 
             if (reviews.data.length === 0) {
               console.log("Not adding pull ready because the PR is not approved yet.")
-              process.exit()
+              process.exit(1)
             }
             let is_approved = false
             if (decision_result.repository.pullRequest.reviewDecision === "APPROVED") {
               is_approved = true
             }
             if (!is_approved) {
               console.log("Not adding pull ready because the PR is not approved yet by sufficient code owners.")
-              process.exit()
+              process.exit(1)
             }
 
             const commits = await github.rest.pulls.listCommits({
@@ -85,11 +100,11 @@ jobs:
               pull_number,
               per_page: 100,
             })
-            // Check that the number of commits in the PR is 1.
-            if (commits.data.length !== 1) {
-              console.log("Not adding pull ready because the PR has more than one commit. Please squash your commits.")
-              process.exit(1)
-            }
+            // Check that the number of commits in the PR is 1. This is not enforced in practice.
+            // if (commits.data.length !== 1) {
+            //  console.log("Not adding pull ready because the PR has more than one commit. Please squash your commits.")
+            //  process.exit(1)
+            // }
             const ref = commits.data.slice(-1)[0].sha
             const checkRuns = await github.rest.checks.listForRef({
               owner,
@@ -98,13 +113,13 @@ jobs:
             })
             if (checkRuns.data.check_runs.length === 0) {
               console.log("Not adding pull ready because no check runs are associated with the last commit: " + ref)
-              process.exit()
+              process.exit(1)
             }
             for (const checkRun of checkRuns.data.check_runs) {
               if (checkRun.name.endsWith(context.job)) continue
               if (checkRun.conclusion !== "success") {
                 console.log("Not adding pull ready because " + checkRun.name + " has not passed yet: " + checkRun.html_url)
-                process.exit()
+                process.exit(1)
               }
             }
             console.log("Adding pull ready label because the PR is approved AND all the check runs have passed")