Update python-app.yml (Test #5)

Deep-Learning-Profiling-Tools · Mar 6, 2024 · cc37e03 · cc37e03
1 parent d87b0ab
commit cc37e03
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
@@ -20,7 +20,6 @@ jobs:
     steps:
     - uses: actions/checkout@v3
       with:
-        ref: tejas3070-patch-1
         path: 'triton_viz'
 
     - name: Set up Python 3.10
@@ -53,6 +52,7 @@ jobs:
         git clone https://github.com/openai/triton.git ~/triton
         cd ~/triton/python
         pip install .
+        pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
 
     - name: Install triton_viz
       run: |
@@ -61,8 +61,7 @@ jobs:
 
     - name: Install main project (triton_viz) dependencies
       run: |
-        cd $GITHUB_WORKSPACE/triton_viz
-        pip install -r requirements.txt
+        pip install pytest pre-commit pyarrow
         
     - name: Lint with pre-commit
       run: |