From 33af948c0a12ff21788cce495d2804cf4a8cd78e Mon Sep 17 00:00:00 2001 From: Tejas Ramesh Date: Fri, 8 Mar 2024 15:23:13 -0500 Subject: [PATCH] Updated setup.py --- .github/workflows/python-app.yml | 4 ++-- requirements.txt | 3 --- setup.py | 3 +++ 3 files changed, 5 insertions(+), 5 deletions(-) delete mode 100644 requirements.txt diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml index e256a09..93b51b3 100644 --- a/.github/workflows/python-app.yml +++ b/.github/workflows/python-app.yml @@ -47,7 +47,7 @@ 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: | @@ -55,7 +55,7 @@ jobs: 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 deleted file mode 100644 index b2e766a..0000000 --- a/requirements.txt +++ /dev/null @@ -1,3 +0,0 @@ -pre-commit -pyarrow -pytest diff --git a/setup.py b/setup.py index a469252..4204c80 100644 --- 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", ], )