Rockset migration - update queue times - set aws permissions in workf…

…low (#5411) Role defined in pytorch-labs/pytorch-gha-infra#441 Related * #5398
pytorch · Jul 9, 2024 · 0f0ff39 · 0f0ff39
1 parent de45e26
commit 0f0ff39
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/.github/workflows/update-queue-times.yml b/.github/workflows/update-queue-times.yml
@@ -11,9 +11,17 @@ defaults:
 jobs:
   update-queue-rockset:
     runs-on: ubuntu-20.04
+    permissions:
+      id-token: write
     steps:
       - uses: actions/checkout@v3
       - run: yarn install --frozen-lockfile
+      - name: configure aws credentials
+        id: aws_creds
+        uses: aws-actions/[email protected]
+        with:
+          role-to-assume: arn:aws:iam::308535385114:role/gha_workflow_update_queue_times
+          aws-region: us-east-1
       - run: yarn node scripts/updateQueueTimes.mjs
         env:
           ROCKSET_API_KEY: ${{ secrets.ROCKSET_API_KEY }}