Add neuronx

huggingface · Aug 3, 2023 · 1dafbf1 · 1dafbf1
1 parent ac8cd69
commit 1dafbf1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test_on_trainium.yml b/.github/workflows/test_on_trainium.yml
@@ -63,7 +63,7 @@ jobs:
       # - name: Install python3.8-venv
       #   run: sudo apt update; sudo apt install -y python3.8-venv
       - name: Install Python dependencies
-        run: pip install .[tests]
+        run: pip install .[tests,neuronx]
       - 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 -k "not test_run_example"