5252 python-version :
5353 - " 3.9"
5454 - " 3.10"
55- - " 3.11"
55+ # TODO put back 3.11 (See blame)
56+ # - "3.11"
5657 - " 3.12"
5758 runner : ["macos-m1-stable"]
5859 fail-fast : false
@@ -81,11 +82,12 @@ jobs:
8182 - " 3.12"
8283 runner : ["windows.4xlarge"]
8384 gpu-arch-type : ["cpu"]
84- include :
85- - python-version : " 3.9"
86- runner : windows.g5.4xlarge.nvidia.gpu
87- gpu-arch-type : cuda
88- gpu-arch-version : " 11.8"
85+ # TODO: put GPU testing back
86+ # include:
87+ # - python-version: "3.9"
88+ # runner: windows.g5.4xlarge.nvidia.gpu
89+ # gpu-arch-type: cuda
90+ # gpu-arch-version: "11.8"
8991 fail-fast : false
9092 uses : pytorch/test-infra/.github/workflows/windows_job.yml@main
9193 permissions :
@@ -109,39 +111,39 @@ jobs:
109111
110112 ./.github/scripts/unittest.sh
111113
112- onnx :
113- uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
114- permissions :
115- id-token : write
116- contents : read
117- with :
118- repository : pytorch/vision
119- test-infra-ref : main
120- script : |
121- set -euo pipefail
122-
123- export PYTHON_VERSION=3.10
124- export GPU_ARCH_TYPE=cpu
125- export GPU_ARCH_VERSION=''
126-
127- ./.github/scripts/setup-env.sh
128-
129- # Prepare conda
130- CONDA_PATH=$(which conda)
131- eval "$(${CONDA_PATH} shell.bash hook)"
132- conda activate ci
133-
134- echo '::group::Install ONNX'
135- pip install --progress-bar=off onnx onnxruntime
136- echo '::endgroup::'
137-
138- echo '::group::Install testing utilities'
139- pip install --progress-bar=off pytest "numpy<2"
140- echo '::endgroup::'
141-
142- echo '::group::Run ONNX tests'
143- pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_onnx.py
144- echo '::endgroup::'
114+ # onnx:
115+ # uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
116+ # permissions:
117+ # id-token: write
118+ # contents: read
119+ # with:
120+ # repository: pytorch/vision
121+ # test-infra-ref: main
122+ # script: |
123+ # set -euo pipefail
124+
125+ # export PYTHON_VERSION=3.10
126+ # export GPU_ARCH_TYPE=cpu
127+ # export GPU_ARCH_VERSION=''
128+
129+ # ./.github/scripts/setup-env.sh
130+
131+ # # Prepare conda
132+ # CONDA_PATH=$(which conda)
133+ # eval "$(${CONDA_PATH} shell.bash hook)"
134+ # conda activate ci
135+
136+ # echo '::group::Install ONNX'
137+ # pip install --progress-bar=off onnx onnxruntime
138+ # echo '::endgroup::'
139+
140+ # echo '::group::Install testing utilities'
141+ # pip install --progress-bar=off pytest "numpy<2"
142+ # echo '::endgroup::'
143+
144+ # echo '::group::Run ONNX tests'
145+ # pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_onnx.py
146+ # echo '::endgroup::'
145147
146148 unittests-extended :
147149 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
0 commit comments