Updated setup.py

Deep-Learning-Profiling-Tools · Mar 8, 2024 · 33af948 · 33af948
1 parent f219dbb
commit 33af948
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
@@ -47,15 +47,15 @@ jobs:
         git clone https://github.com/openai/triton.git
         cd triton/python
         pip install -e .
-        
+
     - name: Install Dependencies if Cache Missed
       if: steps.cache-dependencies.outputs.cache-hit != 'true'
       run: |
         cd triton_viz
         pip install -e .
         pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
         pip uninstall pytorch-triton -y
-        
+
     - name: Lint with pre-commit
       run: |
         cd triton_viz

diff --git a/requirements.txt b/requirements.txt
diff --git a/setup.py b/setup.py
@@ -13,5 +13,8 @@
         "triton",
         "gradio",
         "chalk-diagrams @ git+https://github.com/chalk-diagrams/chalk.git",
+        "pyarrow",
+        "pytest",
+        "pre-commit",
     ],
 )