Fix docker build failure in nighlty (#364)

Fixed two issues:

1. Trigger nm pypi update workflow in stratus automatically after wheels
are uploaded to GCP.
2. Docker build step should only run after the upload step is finished
successfully.

An example run from nm-vllm:
https://github.com/neuralmagic/nm-vllm/actions/runs/9810788569

which triggers stratus nm pypi update workflow run:
https://github.com/neuralmagic/stratus/actions/runs/9810806638

---------

Co-authored-by: dhuangnm <dhuang@MacBook-Pro-2.local>

.github/workflows/nm-build-test.yml

@@ -173,7 +173,7 @@ jobs:
 
     # update docker
     DOCKER:
-        needs: [BUILD]
+        needs: [BUILD, UPLOAD]
         if: ${{ inputs.wf_category != 'REMOTE' }}
         uses: ./.github/workflows/publish-docker.yml
         with:

.github/workflows/nm-upload-assets-to-gcp.yml

@@ -17,6 +17,21 @@ on:
                 type: string
                 required: true
 
+    workflow_dispatch:
+        inputs:
+            label:
+                description: "requested runner label (specifies instance)"
+                type: string
+                required: true
+            timeout:
+                description: "time limit for run in minutes "
+                type: string
+                required: true
+            gitref:
+                description: 'git commit hash or branch name'
+                type: string
+                required: true
+
 jobs:
 
     PUBLISH:
@@ -70,3 +85,16 @@ jobs:
             - name: cp assets
               id: cp-assets
               uses: ./.github/actions/nm-cp-assets/
+
+            - name: trigger stratus nm-pypi update workflow to update nm-pypi index
+              uses: actions/github-script@v6
+              with:
+                github-token: ${{ secrets.CICD_GITHUB_PAT }}
+                script: |
+                  const result = await github.rest.actions.createWorkflowDispatch({
+                    owner: 'neuralmagic',
+                    repo: 'stratus',
+                    workflow_id: 'nm-pypi-update.yml',
+                    ref: 'main'
+                  })
+                  console.log(result)