From e3e41ffe4d7eb50e07bc5af19401729259e4d556 Mon Sep 17 00:00:00 2001
From: Raunak Bhagat <rabhagat31@gmail.com>
Date: Wed, 13 Nov 2024 20:28:40 -0800
Subject: [PATCH] Edit

---
 .github/workflows/benchmark-local.yaml | 25 ++++++++-----------------
 1 file changed, 8 insertions(+), 17 deletions(-)

diff --git a/.github/workflows/benchmark-local.yaml b/.github/workflows/benchmark-local.yaml
index c93ae68ed1..586b68fffe 100644
--- a/.github/workflows/benchmark-local.yaml
+++ b/.github/workflows/benchmark-local.yaml
@@ -105,30 +105,21 @@ jobs:
         sed -i 's|<<COMMIT>>|${{ matrix.commit }}|g' .github/assets/ray.yaml
         sed -i 's|<<WHEEL>>|${{ needs.build.outputs.wheel }}|g' .github/assets/ray.yaml
 
-    - name: Manual breakpoint to check ssh-key
-      uses: lhotari/action-upterm@v1
-
-    - name: Get private ssh-key to boot up ray-cluster
-      id: private-ssh-key
-      uses: aws-actions/aws-secretsmanager-get-secrets@v2
-      with:
-        parse-json-secrets: true
-        secret-ids: |
-          SSH_KEY,ci-github-actions-ray-cluster-key-2
     - name: Write ssh-key to file
       run: |
-        echo "${{ steps.get-secret.outputs.secret }}" >> ~/.ssh/ci-github-actions-ray-cluster-key.pem
+        KEY=aws secretsmanager get-secret-value --secret-id ci-github-actions-ray-cluster-key-2 --query SecretString --output text | jq -r '.private'
+        echo $KEY >> ~/.ssh/ci-github-actions-ray-cluster-key.pem
         chmod 600 ~/.ssh/ci-github-actions-ray-cluster-key.pem
 
     - name: Manual breakpoint to check ssh-key
       uses: lhotari/action-upterm@v1
 
-    # - name: Spin up ray cluster
-    #   run: |
-    #     uv v
-    #     source .venv/bin/activate
-    #     uv pip install ray boto3
-    #     ray up .github/assets/ray.yaml -y
+    - name: Spin up ray cluster
+      run: |
+        uv v
+        source .venv/bin/activate
+        uv pip install ray boto3
+        ray up .github/assets/ray.yaml -y
 
     # - name: Run commit ${{ matrix.commit }}
     #   run: |