add outputs id

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

.github/workflows/pr-sglang.yml

@@ -42,7 +42,7 @@ jobs:
       - codebuild-runner-${{ github.run_id }}-${{ github.run_attempt }}
         fleet:x86-build-runner
     outputs:
-      image_uri: ${{ env.IMAGE_URI }}
+      image_uri: ${{ steps.image_uri_build.outputs.IMAGE_URI }}
     steps:
       - uses: actions/checkout@v5
       - run: .github/scripts/runner_setup.sh
@@ -52,6 +52,7 @@ jobs:
           aws ecr get-login-password --region ${{ secrets.AWS_REGION }} | docker login --username AWS --password-stdin ${{ secrets.AWS_ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com
 
       - name: Resolve image URI for build
+        id: image_uri_build
         run: |
           IMAGE_URI=${{ secrets.AWS_ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/ci:sglang-0.5.5-gpu-py312-cu129-ubuntu22.04-sagemaker-pr-${{ github.event.pull_request.number }}
           echo "Image URI to build: $IMAGE_URI"
@@ -90,17 +91,17 @@ jobs:
       - name: Checkout DLC source
         uses: actions/checkout@v5
 
+      - name: echo container
+        run: |
+          echo "image_uri: ${{ needs.build-sglang-image.outputs.IMAGE_URI }}"
+
       - name: Container pull
         uses: ./.github/actions/container-pull
         with:
           aws_region: ${{ secrets.AWS_REGION }}
           aws_account_id: ${{ secrets.AWS_ACCOUNT_ID }}
           image_uri: ${{ needs.build-sglang-image.outputs.IMAGE_URI }}
 
-      - name: Container pull
-        run: |
-          echo "image_uri: ${{ needs.build-sglang-image.outputs.IMAGE_URI }}"
-
       - name: Setup for SGLang datasets
         run: |
           mkdir -p /tmp/sglang/dataset