resolve comments

Author: YixuanWang-99

…ows/build_and_test_release_candidate.yml .github/workflows/release_pipeline.yml.github/workflows/build_and_test_release_candidate.yml renamed to .github/workflows/release_pipeline.yml .github/workflows/build_and_test_release_candidate.yml renamed to .github/workflows/release_pipeline.yml

@@ -15,32 +15,26 @@
 # This workflow will build and test the MaxText package, and build Docker images
 # for a Release Candidate when a new release tag is created.
 
-name: Build and Test MaxText Release Candidate
+name: MaxText Release Pipeline
 
+# Triggers when a new "release" is published in the GitHub UI
 on:
   release:
     types: [published]
   workflow_dispatch:
-    inputs:
-      target_device:
-        description: 'Target Device (tpu, gpu, or all)'
-        required: false
-        default: 'all'
-        type: choice
-        options:
-          - all
-          - tpu
-          - gpu
 
 permissions:
   contents: read
   issues: write
+  id-token: write
 
 jobs:
   release_approval:
     name: Approve Release Candidate
     if: github.event_name != 'workflow_dispatch'
     runs-on: ubuntu-latest
+    # "release" environment is configured in MaxText repository settings.
+    # This environment requires manual approval before proceeding to the next job.
     environment: release
     steps:
       - name: Acknowledge Approval
@@ -56,7 +50,7 @@ jobs:
     secrets: inherit
 
   build_release_candidate_images:
-    name: Build Release Candidate Images
+    name: ${{ matrix.image_name }}
     needs: [build_and_test_maxtext_package]
     if: |
       always() &&
@@ -83,5 +77,6 @@ jobs:
       workflow: ${{ matrix.workflow }}
       dockerfile: ${{ matrix.dockerfile }}
       maxtext_sha: ${{ github.sha }}
+      include_test_assets: true
     secrets:
       HF_TOKEN: ${{ secrets.HF_TOKEN }}