Update workflow

huggingface · Aug 3, 2023 · ac8cd69 · ac8cd69
1 parent 37ff413
commit ac8cd69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_on_trainium.yml b/.github/workflows/test_on_trainium.yml
@@ -66,9 +66,9 @@ jobs:
         run: pip install .[tests]
       - name: Run tests on neuron cores
         run: |
-          HF_TOKEN_OPTIMUM_NEURON_CI=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} USE_VENV="false" pytest -m "is_trainium_test" tests/test_*.py -k "not test_run_example"
+          HF_TOKEN_OPTIMUM_NEURON_CI=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} USE_VENV="false" pytest -m "is_trainium_test" tests -k "not test_run_example"
       - name: Run staging tests on neuron cores 
-        run: HUGGINGFACE_CO_STAGING=1 pytest -m "is_trainium_test and is_staging_test" tests/test_*.py
+        run: HUGGINGFACE_CO_STAGING=1 pytest -m "is_trainium_test and is_staging_test" tests
   stop-runner:
     name: Stop self-hosted EC2 runner
     needs: