From 7f90c494bf18b03c439085c676d12d02a7ea841e Mon Sep 17 00:00:00 2001
From: dhuangnm <dhuang@MacBook-Pro-2.local>
Date: Thu, 12 Sep 2024 13:30:59 -0400
Subject: [PATCH] use secret for bucket

---
 .github/workflows/build-test.yml | 2 +-
 .github/workflows/build.yml      | 6 +++---
 .github/workflows/upload.yml     | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build-test.yml b/.github/workflows/build-test.yml
index acc961c4..cbb74e7e 100644
--- a/.github/workflows/build-test.yml
+++ b/.github/workflows/build-test.yml
@@ -60,7 +60,7 @@ jobs:
         needs: TEST
         uses: ./.github/workflows/upload.yml
         with:
-            label: k8s-util
+            label: ${{ inputs.build_label }}
             timeout: 40
             run_id: ${{ github.run_id }}
             push_to_pypi: ${{ inputs.push_to_pypi }}
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index c1af0680..9ab4774a 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -58,7 +58,7 @@ jobs:
             id-token: 'write'
 
         outputs:
-            assets_location: gs://compressed-tensors/assets/${{ github.run_id }}
+            assets_location: ${{ secrets.GCP_BUILD_ML_ASSETS2 }}/assets/${{ github.run_id }}
             run_id: ${{ github.run_id }}
             whl: ${{ steps.build.outputs.whl }}
             tarfile: ${{ steps.build.outputs.tarfile }}
@@ -114,8 +114,8 @@ jobs:
 
             - name: copy whl and source distribution
               run: |
-                  gcloud storage cp dist/${{ steps.build.outputs.whlname }} gs://compressed-tensors/assets/${{ github.run_id }}/${{ steps.build.outputs.whlname }}
-                  gcloud storage cp dist/${{ steps.build.outputs.tarname }} gs://compressed-tensors/assets/${{ github.run_id }}/${{ steps.build.outputs.tarname }}
+                  gcloud storage cp dist/${{ steps.build.outputs.whlname }} ${{ secrets.GCP_BUILD_ML_ASSETS2 }}/assets/${{ github.run_id }}/${{ steps.build.outputs.whlname }}
+                  gcloud storage cp dist/${{ steps.build.outputs.tarname }} ${{ secrets.GCP_BUILD_ML_ASSETS2 }}/assets/${{ github.run_id }}/${{ steps.build.outputs.tarname }}
 
             - name: upload whl
               uses: actions/upload-artifact@v4
diff --git a/.github/workflows/upload.yml b/.github/workflows/upload.yml
index ab945816..62648ba1 100644
--- a/.github/workflows/upload.yml
+++ b/.github/workflows/upload.yml
@@ -84,7 +84,7 @@ jobs:
             - name: download assets
               uses: neuralmagic/nm-actions/actions/gcp-download-assets@v1.1.0
               with:
-                bucket_source: ${{ secrets.GCP_BUILD_ASSETS }}
+                bucket_source: ${{ secrets.GCP_BUILD_ML_ASSETS2 }}
                 run_id: ${{ inputs.run_id }}
 
             - name: find whl