use conda

Author: Allen Wang

.github/workflows/docs.yml

@@ -18,6 +18,14 @@ jobs:
         uses: actions/checkout@v4
         with:
           fetch-depth: 0
+      - name: Setup conda env
+        uses: conda-incubator/setup-miniconda@v2
+        with:
+          auto-update-conda: true
+          miniconda-version: "latest"
+          activate-environment: test
+          python-version: '3.10'
+          auto-activate: false
       - name: Update pip
         shell: bash -l {0}
         run: python -m pip install --upgrade pip
@@ -26,7 +34,7 @@ jobs:
         run: curl -LsSf https://astral.sh/uv/install.sh | sh
       - name: Install torchforge
         shell: bash -l {0}
-        run: uv sync --all-extras && source .venv/bin/activate
+        run: uv sync
       - name: Install docs dependencies
         shell: bash -l {0}
         run: python -m pip install -r docs/requirements.txt

.github/workflows/gpu_test.yaml

@@ -28,12 +28,19 @@ jobs:
     steps:
       - name: Check out repo
         uses: actions/checkout@v4
-      - name: Install uv
-        run: curl -LsSf https://astral.sh/uv/install.sh | sh
+      - name: Setup conda env
+        uses: conda-incubator/setup-miniconda@v2
+        with:
+          auto-update-conda: true
+          miniconda-version: "latest"
+          activate-environment: test
+          python-version: ${{ matrix.python-version }}
+      - name: Update pip
+        run: python -m pip install --upgrade pip
       - name: Install torchforge
-        run: uv sync --all-extras && source .venv/bin/activate
+        run: uv sync
       - name: Run unit tests with coverage
         # TODO add all tests
-        run: uv run pytest tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv
+        run: pytest tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv
       - name: Upload Coverage to Codecov
         uses: codecov/codecov-action@v3