From 23e77887f1ef4b006cc20c0cd2c66b28df774af8 Mon Sep 17 00:00:00 2001
From: Tejas Ramesh <tramesh2@gmu.edu>
Date: Wed, 6 Mar 2024 18:07:40 -0500
Subject: [PATCH] Updated requirements.txt

---
 .github/workflows/python-app.yml | 8 ++++----
 Examples/conftest.py             | 3 ++-
 requirements.txt                 | 5 +----
 triton_viz/interface.py          | 3 +--
 4 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
index 0e2a245..bb4783b 100644
--- a/.github/workflows/python-app.yml
+++ b/.github/workflows/python-app.yml
@@ -20,12 +20,12 @@ jobs:
     - uses: actions/checkout@v3
       with:
         path: 'triton_viz'
-        
+
     - name: Set up Python 3.10
       uses: actions/setup-python@v3
       with:
         python-version: "3.10"
-        
+
     - name: Create virtual environment
       run: |
         python -m venv triton_env
@@ -57,12 +57,12 @@ jobs:
         pip install .
         cd triton_viz
         pip install -r triton_viz/requirements.txt
-        
+
     - name: Lint with pre-commit
       run: |
         cd triton_viz
         pre-commit run --all-files
-    
+
     - name: Test with pytest
       run: |
         source triton_env/bin/activate
diff --git a/Examples/conftest.py b/Examples/conftest.py
index ae18223..c326b6f 100644
--- a/Examples/conftest.py
+++ b/Examples/conftest.py
@@ -1,7 +1,8 @@
 import pytest
 from triton_viz.interpreter import record_builder
 
-@pytest.fixture(autouse=True, scope='function')
+
+@pytest.fixture(autouse=True, scope="function")
 def clear_cache():
     yield
     record_builder.reset()
diff --git a/requirements.txt b/requirements.txt
index 4cfd45d..b2e766a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,3 @@
-pyarrow
-gradio
-numpy
 pre-commit
+pyarrow
 pytest
---pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
diff --git a/triton_viz/interface.py b/triton_viz/interface.py
index 6f7a871..a60c1f7 100644
--- a/triton_viz/interface.py
+++ b/triton_viz/interface.py
@@ -3,7 +3,6 @@
 import tempfile
 
 
-
 def launch():
     cache = {}
     program_records, tt = triton_viz.collect_grid()
@@ -19,7 +18,7 @@ def launch():
                 s1 = gr.Slider(0, m[0], value=0, step=1, label="Program Id 0")
                 s2 = gr.Slider(0, m[1], value=0, step=1, label="Program Id 1")
                 s3 = gr.Slider(0, m[2], value=0, step=1, label="Program Id 2")
-        
+
         def update(inp):
             a = inp[s1]
             b = inp[s2]