diff --git a/.github/workflows/unit_test.yaml b/.github/workflows/unit_test.yaml index ed4cbc5c3..d9e5dbe06 100644 --- a/.github/workflows/unit_test.yaml +++ b/.github/workflows/unit_test.yaml @@ -27,6 +27,11 @@ jobs: run: python -m pip install torch==2.9.0.dev20250826 --extra-index-url https://download.pytorch.org/whl/nightly/cpu - name: Install monarch run: python -m pip install monarch-no-torch==0.1.0.dev20250826 --find-links assets/ci + - name: Install torchstore + run: | + eval "$(ssh-agent -s)" + ssh-add - <<< '${{ secrets.FORGE_GITHUB_CI_FOR_TORCHSTORE }}' + python -m pip install git+ssh://git@github.com/meta-pytorch/torchstore.git - name: Install dependencies run: python -m pip install --no-build-isolation -e ".[dev]" - name: Run unit tests with coverage