use install script in GHA workflows

Author: ebsmothers

.github/workflows/gpu_test.yaml

@@ -40,21 +40,8 @@ jobs:
           python-version: ${{ matrix.python-version }}
       - name: Update pip
         run: python -m pip install --upgrade pip
-      - name: Install pinned torch nightly
-        run: python -m pip install --pre torch==2.9.0.dev20250905 --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cu129
-      - name: Download and install vLLM and its dependencies
-        # TODO: this honestly could not be hackier if I tried
-        run: |
-          python -m pip install -r .github/packaging/vllm_reqs_12_9.txt
-          python -m pip install vllm==0.10.1.dev0+g6d8d0a24c.d20251009.cu129 --no-cache-dir --index-url https://download.pytorch.org/whl/preview/forge
-      - name: Install Monarch
-        run: pip install torchmonarch==0.1.0rc1
-      - name: Install torchtitan and torchstore
-        run: |
-          python -m pip install git+https://github.com/pytorch/torchtitan.git
-          python -m pip install git+https://github.com/meta-pytorch/torchstore.git
-      - name: Install dependencies
-        run: python -m pip install --no-build-isolation -e ".[dev]"
+      - name: Install forge
+        run: ./scripts/install.sh
       - name: Run unit tests with coverage
         # TODO add all tests
         run: |

.github/workflows/unit_test.yaml

@@ -25,18 +25,8 @@ jobs:
           python-version: ${{ matrix.python-version }}
       - name: Update pip
         run: python -m pip install --upgrade pip
-      - name: Install pytorch
-        run: python -m pip install torch==2.9.0.dev20250826 --extra-index-url https://download.pytorch.org/whl/nightly/cpu
-      - name: Install monarch
-        run: pip install assets/ci/monarch_no_torch-0.1.0.dev20251010-py3-none-any.whl
-      - name: Install torchstore
-        run: pip install assets/wheels/torchstore-0.1.0-py3-none-any.whl
-      - name: Install torchtitan
-        run: |
-          pip install assets/wheels/torchtitan-0.1.0-py3-none-any.whl
-          pip install tyro
-      - name: Install dependencies
-        run: python -m pip install --no-build-isolation -e ".[dev]"
+      - name: Install forge
+        run: ./scripts/install.sh
       - 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

@@ -281,7 +281,7 @@ main() {
     pip install "$WHEEL_DIR"/*.whl
 
     log_info "Installing Forge from source..."
-    pip install -e .
+    pip install -e ".[dev]"
 
     # Set up environment
     log_info "Setting up environment..."