fix workflows

NVIDIA · Dec 2, 2023 · a7dc646 · a7dc646
1 parent a0cded0
commit a7dc646
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/nightly-rosetta-pax-build.yaml b/.github/workflows/nightly-rosetta-pax-build.yaml
@@ -99,6 +99,7 @@ jobs:
     needs: [metadata, amd64, arm64]
     uses: ./.github/workflows/_publish_badge.yaml
     if: always()
+    with:
       ENDPOINT_FILENAME: 'rosetta-pax-build-status.json'
       PUBLISH: ${{ needs.metadata.outputs.PUBLISH == 'true' }}
       SCRIPT: |
@@ -152,7 +153,7 @@ jobs:
         type=raw,value=nightly-${{ needs.metadata.outputs.BUILD_DATE }},priority=900
 
   publish-test-badge:
-    needs: [metadata, publish-build-badge, test-amd64]
+    needs: [metadata, amd64, arm64, test-amd64]
     uses: ./.github/workflows/_publish_badge.yaml
     if: always()
     secrets: inherit

diff --git a/.github/workflows/nightly-rosetta-t5x-build-test.yaml b/.github/workflows/nightly-rosetta-t5x-build-test.yaml
@@ -178,7 +178,7 @@ jobs:
     secrets: inherit
 
   publish-test-badge:
-    needs: [metadata, publish-build-badge, test-unit-amd64, test-t5x-amd64, test-vit-amd64]
+    needs: [metadata, amd64, arm64, test-unit-amd64, test-t5x-amd64, test-vit-amd64]
     uses: ./.github/workflows/_publish_badge.yaml
     if: always()
     secrets: inherit