From 7b44de590101a659e8aab92f9384d5934677098f Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 13 Mar 2024 16:53:40 -0700
Subject: [PATCH] Update load_test.yml

---
 .github/workflows/load_test.yml | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/.github/workflows/load_test.yml b/.github/workflows/load_test.yml
index eca31e23a28..6cf6bc6e711 100644
--- a/.github/workflows/load_test.yml
+++ b/.github/workflows/load_test.yml
@@ -10,14 +10,10 @@ jobs:
         uses: actions/checkout@v1
       - name: Run Load Test
         id: locust_run
-        uses: apardo04/locust-github-action@master
+        uses: BerriAI/locust-github-action@master
         with:
           LOCUSTFILE: ".github/workflows/locustfile.py"
           URL:  "https://litellm-api.up.railway.app/"
           USERS: "5"
           RATE: "5"
           RUNTIME: "10s"
-      - name: Echo Load Test Output
-        run: |
-          echo "Load Test Output: ${{ steps.locust_run.outputs }}"
-          echo "Load Test Output: ${{ steps.locust_run.outputs.logs }}"