Add 3.14 to unittest jobs matrix

.github/workflows/tests.yml

@@ -15,15 +15,14 @@ jobs:
       matrix:
         python-version:
           - "3.10"
-          - "3.11"
-          - "3.12"
+          - "3.14"
         runner: ["linux.12xlarge"]
         gpu-arch-type: ["cpu"]
         include:
           - python-version: "3.10"
             runner: linux.g5.4xlarge.nvidia.gpu
             gpu-arch-type: cuda
-            gpu-arch-version: "12.6"
+            gpu-arch-version: "12.8"
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions:
@@ -80,9 +79,7 @@ jobs:
       matrix:
         python-version:
           - "3.10"
-          # TODO put back 3.11 (See blame)
-          # - "3.11"
-          - "3.12"
+          - "3.14"
         runner: ["macos-m1-stable"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
@@ -105,16 +102,15 @@ jobs:
       matrix:
         python-version:
           - "3.10"
-          - "3.11"
-          - "3.12"
+          - "3.14"
         runner: ["windows.4xlarge"]
         gpu-arch-type: ["cpu"]
         # TODO: put GPU testing back
         # include:
-        #   - python-version: "3.9"
+        #   - python-version: "3.10"
         #     runner: windows.g5.4xlarge.nvidia.gpu
         #     gpu-arch-type: cuda
-        #     gpu-arch-version: "11.8"
+        #     gpu-arch-version: "12.8"
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
     permissions:
@@ -138,40 +134,6 @@ jobs:
 
         ./.github/scripts/unittest.sh
 
-  # onnx:
-  #   uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
-  #   permissions:
-  #     id-token: write
-  #     contents: read
-  #   with:
-  #     repository: pytorch/vision
-  #     test-infra-ref: main
-  #     script: |
-  #       set -euo pipefail
-
-  #       export PYTHON_VERSION=3.10
-  #       export GPU_ARCH_TYPE=cpu
-  #       export GPU_ARCH_VERSION=''
-
-  #       ./.github/scripts/setup-env.sh
-
-  #       # Prepare conda
-  #       CONDA_PATH=$(which conda)
-  #       eval "$(${CONDA_PATH} shell.bash hook)"
-  #       conda activate ci
-
-  #       echo '::group::Install ONNX'
-  #       pip install --progress-bar=off onnx onnxruntime
-  #       echo '::endgroup::'
-
-  #       echo '::group::Install testing utilities'
-  #       pip install --progress-bar=off pytest "numpy<2"
-  #       echo '::endgroup::'
-
-  #       echo '::group::Run ONNX tests'
-  #       pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_onnx.py
-  #       echo '::endgroup::'
-
   unittests-extended:
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     permissions: