dev

Author: Allen Wang

.github/workflows/gpu_test.yaml

@@ -39,7 +39,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install torchforge
         # run: ./scripts/install.sh
-        run: pip install uv && uv pip install .
+        run: pip install uv && uv pip install . && uv pip install .[dev]
       - name: Run unit tests with coverage
         # TODO add all tests
         run: pytest tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv

scripts/install.sh

@@ -196,7 +196,7 @@ main() {
     uv pip install -r .github/packaging/vllm_reqs_12_8.txt
     uv pip install six
     uv pip install "setuptools<80"
-    uv python -m pip install vllm --no-cache-dir --index-url https://download.pytorch.org/whl/preview/forge
+    python -m pip install vllm --no-cache-dir --index-url https://download.pytorch.org/whl/preview/forge
 
     # Install monarch
     uv pip install torchmonarch==$MONARCH_VERSION