[CI] make AWS tests reuse torch + triton installations from build job

Author: latentCall145

.github/workflows/python-app.yml

@@ -61,39 +61,12 @@ jobs:
         cd triton_viz
         python -m pytest tests
 
-  nki-tests:
-    name: AWS Neuron / NKI
-    runs-on: ubuntu-22.04
-    needs: build
-    env:
-      TRITON_INTERPRET: "1"
-      TRITON_VERSION: "3.4.0"
-      PIP_EXTRA_INDEX_URL: https://pip.repos.neuron.amazonaws.com
-
-    steps:
-    - 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: Install AWS Neuron Python packages
-      run: |
-        python -m pip install neuronx-cc==2.* --extra-index-url https://pip.repos.neuron.amazonaws.com
-
-    - name: Install Triton
-      run: |
-        python -m pip install triton==${{ env.TRITON_VERSION }}
-
     - name: Install Triton-Viz with NKI extras
       run: |
         cd triton_viz
-        python -m pip install -e .[nki]
+        pip install -e .[nki]
 
-    - name: Run NKI pytest suite
+    - name: Run full (Triton + NKI) pytest suite
       run: |
         cd triton_viz
-        python -m pytest tests/nki -m ""
+        python -m pytest tests -m ""