add test matrix with uv

Author: apoorvkh

.github/workflows/main.yml

@@ -26,62 +26,40 @@ jobs:
       - run: uv run --frozen pyright
         if: success() || failure()
 
+  ##
+
+  get-pytorch-versions:
+    runs-on: ubuntu-latest
+    outputs:
+      versions: ${{ steps.get-pytorch-versions.outputs.versions }}
+    steps:
+      - name: Get PyTorch versions
+        id: get-pytorch-versions
+        run: |
+          VERSIONS=$(
+            curl -s https://pypi.org/pypi/torch/json | jq -r '.releases | keys[]' |
+            # remove versions <2.0; strip "patch" from versions
+            grep -v '^1\.' | grep -E '\.[0]+$' | sort -V | sed 's/\.0$//' |
+            # to JSON array
+            jq -R . | jq -sc .
+          )
+          echo "versions=$VERSIONS" >> $GITHUB_OUTPUT
+          # e.g. ["2.0","2.1","2.2","2.3","2.4"]
+
   test:
     runs-on: ubuntu-latest
+    needs: get-pytorch-versions
+    strategy:
+      fail-fast: false
+      matrix:
+        python: ["3.9", "3.10", "3.11", "3.12"] 
+        pytorch: ${{fromJson(needs.get-pytorch-versions.outputs.versions)}}
+
     steps:
       - uses: actions/checkout@v4
       - uses: astral-sh/[email protected]
         with:
           version: "0.5.0"
-          python-version-file: ".python-version"
-          enable-cache: true
-      - run: uv sync
-      - run: uv run --frozen pytest tests/test_ci.py
-        if: success() || failure()
-
-  ##
-
-  # get-pytorch-versions:
-  #   runs-on: ubuntu-latest
-  #   outputs:
-  #     versions: ${{ steps.get-pytorch-versions.outputs.versions }}
-  #   steps:
-  #     - name: Get PyTorch versions
-  #       id: get-pytorch-versions
-  #       run: |
-  #         VERSIONS=$(
-  #           curl -s https://pypi.org/pypi/torch/json | jq -r '.releases | keys[]' |
-  #           # remove versions <2.0; strip "patch" from versions
-  #           grep -v '^1\.' | grep -E '\.[0]+$' | sort -V | sed 's/\.0$//' |
-  #           # to JSON array
-  #           jq -R . | jq -sc .
-  #         )
-  #         echo "versions=$VERSIONS" >> $GITHUB_OUTPUT
-  #         # e.g. ["2.0","2.1","2.2","2.3","2.4"]
-
-  # test:
-  #   runs-on: ubuntu-latest
-  #   needs: get-pytorch-versions
-  #   strategy: 
-  #     fail-fast: false
-  #     matrix:
-  #       python: ["3.9", "3.10", "3.11", "3.12"] 
-  #       pytorch: ${{fromJson(needs.get-pytorch-versions.outputs.versions)}}
-
-  #   steps:
-  #     - uses: actions/checkout@v4
-
-  #     - name: Inject (python / pytorch) test deps into pixi.toml / pyproject.toml
-  #       run: |
-  #         sed -i 's/python = "3.9"/python = "${{ matrix.python }}.*"/' pixi.toml
-  #         sed -i 's/torch>=2.0.0/torch~=${{ matrix.pytorch }}/' pyproject.toml
-
-  #     - uses: prefix-dev/[email protected]
-  #       with:
-  #         pixi-version: v0.27.1
-  #         locked: false
-  #         frozen: false
-  #         cache: false
-  #         environments: default
-  #         activate-environment: default
-  #     - run: pytest tests/test_ci.py
+    - if: contains('2.0,2.1,2.2', matrix.pytorch)
+      run: echo "NUMPY_VERSION='--with numpy<2'" >> $GITHUB_ENV
+    - run: uv run --python ${{ matrix.python }} --with torch~=${{ matrix.pytorch }} ${{ env.NUMPY_VERSION }} pytest --verbose tests/test_ci.py