correct test paths

axonn-ai · Oct 5, 2023 · 4b76f70 · 4b76f70
1 parent cda74aa
commit 4b76f70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/nvidia-tests.yaml b/.github/workflows/nvidia-tests.yaml
@@ -28,7 +28,7 @@ jobs:
         export G_data=$(( 2 / G_inter ))
         export memopt=${{ matrix.memopt }}
         echo "training with G_inter = ${G_inter}, G_data = $(( 2 / G_inter  )) ${{ matrix.memopt }}" 
-        mpirun -n 2 pytest --with-mpi ./tests/test_vit.py 
+        mpirun -n 2 pytest --with-mpi test_vit.py 
     - name: Uninstall AxoNN
       run: |
         pip uninstall --yes axonn
@@ -43,7 +43,7 @@ jobs:
         pip install -r requirements.txt
     - name: Run unit intra-layer unit tests 
       run: |
-        mpirun -n 2 pytest --with-mpi ./tests/test_intra_layer_fc.py 
+        mpirun -n 2 pytest --with-mpi test_intra_layer_fc.py 
     - name: Uninstall AxoNN
       run: |
         pip uninstall --yes axonn