fix?

Author: Allen Wang

.github/workflows/gpu_test.yaml

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