all extras

Author: Allen Wang

.github/workflows/docs.yml

@@ -31,7 +31,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install torchforge
         shell: bash -l {0}
-        run: pip install uv && uv sync
+        run: pip install uv && uv sync --all-extras
       - name: Install docs dependencies
         shell: bash -l {0}
         run: python -m pip install -r docs/requirements.txt

.github/workflows/gpu_test.yaml

@@ -38,7 +38,7 @@ jobs:
       - name: Update pip
         run: python -m pip install --upgrade pip
       - name: Install torchforge
-        run: pip install uv && uv sync
+        run: pip install uv && uv sync --all-extras
       - name: Run unit tests with coverage
         # TODO add all tests
         run: pytest tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv