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", ], )