diff --git a/.github/workflows/unit_test_4gpu.yaml b/.github/workflows/unit_test_4gpu.yaml
index a24db9c1..16a71e65 100644
--- 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
index 8c9af57a..2e55af2b 100644
--- 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