pip install modules for unit testing

pytorch · May 17, 2024 · f7a5832 · f7a5832
1 parent 261901c
commit f7a5832
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.github/workflows/unit_test_4gpu.yaml b/.github/workflows/unit_test_4gpu.yaml
@@ -30,5 +30,6 @@ jobs:
 
         python -m pip install --force-reinstall --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
         python -m pip install --pre torchdata --index-url https://download.pytorch.org/whl/nightly/
+        pip install -e ".[dev]"
         mkdir artifacts-to-be-uploaded
         python ./test_runner.py artifacts-to-be-uploaded
diff --git a/.github/workflows/unit_test_cpu.yaml b/.github/workflows/unit_test_cpu.yaml
@@ -24,4 +24,5 @@ jobs:
 
         pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
         pip install --pre torchdata --index-url https://download.pytorch.org/whl/nightly
+        pip install -e ".[dev]"
         pytest test --cov=. --cov-report=xml --durations=20 -vv