From 8479b25088a646d2648e067435f6e92f20e50843 Mon Sep 17 00:00:00 2001
From: Rachneet <rachneetsachdeva@yahoo.com>
Date: Fri, 3 Jan 2025 13:14:33 +0100
Subject: [PATCH] chore: update CI

---
 .github/workflows/tests.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 9b3f8fb..d700849 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -30,7 +30,7 @@ jobs:
           
       - name: Install dependencies
         run: |
-          pip install torch==2.4.0 packaging
+          pip install torch==2.4.0 packaging wheel
           pip install flash-attn==2.6.3 --no-build-isolation
           pip install .
           pip install -r requirements-dev.txt