ci: move to uv for setup

jcmgray · Oct 21, 2024 · 2ac2d1c · 2ac2d1c
1 parent 9acdbde
commit 2ac2d1c
Show file tree

Hide file tree

Showing 9 changed files with 62 additions and 76 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -35,21 +35,27 @@ jobs:
           env: tensorflow
 
     steps:
-    - uses: actions/checkout@v4
-
-    - name: Install micromamba
-      uses: mamba-org/setup-micromamba@v1
-      with:
-        environment-file: ci/requirements/py-${{ matrix.env }}.yml
-        environment-name: test-env
-        create-args: >-
-          python=${{ matrix.python-version }}
-        cache-environment: true
-
-    - name: Test with pytest
-      run: pytest --cov=autoray tests/ --cov-report=xml tests
-
-    - name: Report to codecov
-      uses: codecov/codecov-action@v4
-      with:
-        token: ${{ secrets.CODECOV_TOKEN }}
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          version: "0.4.20"
+
+      - name: Install python
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Setup uv project
+        run: uv sync
+
+      - name: Install dependencies
+        run: uv pip install -r ci/requirements/${{ matrix.env }}.txt
+
+      - name: Test with pytest
+        run: uv run pytest --cov=autoray tests/ --cov-report=xml tests
+
+      - name: Report to codecov
+        uses: codecov/codecov-action@v4
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+
diff --git a/ci/requirements/base.txt b/ci/requirements/base.txt
@@ -0,0 +1,11 @@
+coverage
+dask
+matplotlib
+networkx
+numba>=0.56
+numpy
+opt_einsum
+pytest
+pytest-cov
+scipy
+sparse
diff --git a/ci/requirements/jax.txt b/ci/requirements/jax.txt
@@ -0,0 +1,7 @@
+coverage
+jax
+matplotlib
+numpy
+opt_einsum
+pytest
+pytest-cov
diff --git a/ci/requirements/py-base.yml b/ci/requirements/py-base.yml
diff --git a/ci/requirements/py-jax.yml b/ci/requirements/py-jax.yml
diff --git a/ci/requirements/py-tensorflow.yml b/ci/requirements/py-tensorflow.yml
diff --git a/ci/requirements/py-torch.yml b/ci/requirements/py-torch.yml
diff --git a/ci/requirements/tensorflow.txt b/ci/requirements/tensorflow.txt
@@ -0,0 +1,9 @@
+coverage
+matplotlib
+networkx
+numpy
+opt_einsum
+pytest
+pytest-cov
+scipy
+tensorflow
diff --git a/ci/requirements/torch.txt b/ci/requirements/torch.txt
@@ -0,0 +1,11 @@
+coverage
+matplotlib
+networkx
+numpy
+opt_einsum
+pytest
+pytest-cov
+scipy
+
+--index-url https://download.pytorch.org/whl/cpu
+torch