Updated requirements.txt

.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

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()

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

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]