Update docker build workflow

Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>

Author: thomasdhc

.github/workflows/_build_container.yml

@@ -20,9 +20,19 @@ on:
         description: "The name of the repo to build container"
         required: true
         type: string
+      runner_name:
+        description: "Name of the runner"
+        required: true
+        type: string
+      environment_name:
+        description: "GitHub environment to use for this build"
+        required: true
+        type: string
 
 jobs:
   docker-build:
+    runs-on: ${{ inputs.runner_name }}
+    environment: ${{ inputs.environment_name }}
     steps:
       - name: Checkout
         uses: actions/checkout@v4

.github/workflows/cicd-main.yml

@@ -131,20 +131,18 @@ jobs:
 
   cicd-container-build:
     needs: [pre-flight, cicd-wait-in-queue]
-    runs-on: self-hosted-nemo
     if: |
       (
         success()
         || needs.pre-flight.outputs.is_ci_workload == 'true'
         || needs.pre-flight.outputs.force_run_all == 'true'
       )
       && !cancelled()
-    environment: nemo-ci
-    steps:
-      - name: Build docker container
-        uses: ./.github/workflows/_build_container.yml
-        with:
-          repo_name: "Automodel"
+    uses: ./.github/workflows/_build_container.yml
+    with:
+      environment_name: "nemo-ci"
+      repo_name: "Automodel"
+      runner_name: self-hosted-nemo
 
   cicd-unit-tests:
     strategy: