From cda97a5cfb59e37d04224a0dd7967f538b727938 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Wed, 11 Dec 2024 21:55:48 +0100
Subject: [PATCH] misc(ci): lets do differently

---
 .github/workflows/build_trtllm.yaml | 13 ++++++-------
 .github/workflows/s3-cache.yaml     |  1 -
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
index 8d9973b129c..c74282bfff9 100644
--- a/.github/workflows/build_trtllm.yaml
+++ b/.github/workflows/build_trtllm.yaml
@@ -24,20 +24,19 @@ permissions:
   id-token: write # Required to authenticate via OIDC.
   
 jobs:
-  call-setup-aws-s3-cache:
-    uses: ./.github/workflows/s3-cache.yaml  # TODO: Change @s3-cache to @main when merged
-    with:
-      runner-group: aws-highmemory-32-plus-priv
-    secrets: inherit
-
   build:
     runs-on:
       group: aws-highmemory-32-plus-priv
-    needs: call-setup-aws-s3-cache
     steps:
       - uses: actions/checkout@v4
       - uses: actions-rust-lang/setup-rust-toolchain@v1
 
+      - name: "Configure AWS Credentials"
+        uses: aws-actions/configure-aws-credentials@v4
+        with:
+          aws-region: us-east-1
+          role-to-assume: ${{ secrets.AWS_ROLE_GITHUB_TGI_TEST }}
+
       - name: "Install sccache"
         run: cargo install sscache
 
diff --git a/.github/workflows/s3-cache.yaml b/.github/workflows/s3-cache.yaml
index dd50a4308a0..7e282d8ed03 100644
--- a/.github/workflows/s3-cache.yaml
+++ b/.github/workflows/s3-cache.yaml
@@ -8,7 +8,6 @@ on:
         required: true
         type: string
 
-
 permissions:
   contents: read # Required to check out repository.
   id-token: write # Required to authenticate via OIDC.