diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
index 5bc472b..7888675 100644
--- a/.github/workflows/python-app.yml
+++ b/.github/workflows/python-app.yml
@@ -20,7 +20,6 @@ jobs:
     steps:
     - uses: actions/checkout@v3
       with:
-        ref: tejas3070-patch-1
         path: 'triton_viz'
         
     - name: Set up Python 3.10
@@ -53,6 +52,7 @@ jobs:
         git clone https://github.com/openai/triton.git ~/triton
         cd ~/triton/python
         pip install .
+        pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
 
     - name: Install triton_viz
       run: |
@@ -61,8 +61,7 @@ jobs:
 
     - name: Install main project (triton_viz) dependencies
       run: |
-        cd $GITHUB_WORKSPACE/triton_viz
-        pip install -r requirements.txt
+        pip install pytest pre-commit pyarrow
         
     - name: Lint with pre-commit
       run: |