[sharktank] Extend to test with Python 3.12

nod-ai · Dec 13, 2024 · 449ae4c · 449ae4c
1 parent 0bec8f1
commit 449ae4c
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/.github/workflows/ci-sharktank.yml b/.github/workflows/ci-sharktank.yml
@@ -26,8 +26,12 @@ jobs:
     name: "Unit Tests and Type Checking"
     strategy:
       matrix:
-        version: [3.11]
+        python-version: ["3.11", "3.12"]
+        torch-versiion: ["2.3.0+cpu", "2.4.1+cpu", "2.5.1+cpu"]
         os: [ubuntu-24.04, windows-2022]
+        exclude:
+          - python-version: "3.12"
+            torch-version: "2.3.0+cpu"
       fail-fast: false
     runs-on: ${{matrix.os}}
     defaults:
@@ -42,7 +46,7 @@ jobs:
         id: setup_python
         uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b # v5.3.0
         with:
-          python-version: ${{matrix.version}}
+          python-version: ${{matrix.python-version}}
 
       - name: Cache Pip Packages
         uses: actions/cache@6849a6489940f00c2f30c0fb92c6274307ccb58a # v4.1.2
@@ -77,7 +81,7 @@ jobs:
     name: "Data-dependent Tests"
     strategy:
       matrix:
-        version: [3.11]
+        python-version: [3.11]
         runs-on: [llama-mi300x-3]
       fail-fast: false
     runs-on: ${{matrix.runs-on}}
@@ -94,7 +98,7 @@ jobs:
         id: setup_python
         uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b # v5.3.0
         with:
-          python-version: ${{matrix.version}}
+          python-version: ${{matrix.python-version}}
 
       - name: Create Python venv
         run: python -m venv ${VENV_DIR}