Skip to content

Commit f130f2e

Browse files
NicolasHugNicolasHug
committed
Install torchvision-extra-decoders for unittests
1 parent cbbe0dc commit f130f2e

File tree

2 files changed

+130
-125
lines changed

2 files changed

+130
-125
lines changed

‎.github/scripts/setup-env.sh‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,11 @@ echo '::group::Install TorchVision'
102102
python setup.py develop
103103
echo '::endgroup::'
104104

105+
echo '::group::Install torchvision-extra-decoders'
106+
# This can be done after torchvision was built
107+
pip install torchvision-extra-decoders
108+
echo '::endgroup::'
109+
105110
echo '::group::Collect environment information'
106111
conda list
107112
python -m torch.utils.collect_env

‎.github/workflows/tests.yml‎

Lines changed: 125 additions & 125 deletions
Original file line numberDiff line numberDiff line change
@@ -43,128 +43,128 @@ jobs:
4343
4444
./.github/scripts/unittest.sh
4545
46-
unittests-macos:
47-
strategy:
48-
matrix:
49-
python-version:
50-
- "3.9"
51-
- "3.10"
52-
- "3.11"
53-
- "3.12"
54-
runner: ["macos-m1-stable"]
55-
fail-fast: false
56-
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
57-
with:
58-
repository: pytorch/vision
59-
timeout: 240
60-
runner: ${{ matrix.runner }}
61-
test-infra-ref: main
62-
script: |
63-
set -euo pipefail
64-
65-
export PYTHON_VERSION=${{ matrix.python-version }}
66-
export GPU_ARCH_TYPE=cpu
67-
export GPU_ARCH_VERSION=''
68-
69-
${CONDA_RUN} ./.github/scripts/unittest.sh
70-
71-
unittests-windows:
72-
strategy:
73-
matrix:
74-
python-version:
75-
- "3.9"
76-
- "3.10"
77-
- "3.11"
78-
- "3.12"
79-
runner: ["windows.4xlarge"]
80-
gpu-arch-type: ["cpu"]
81-
include:
82-
- python-version: "3.9"
83-
runner: windows.g5.4xlarge.nvidia.gpu
84-
gpu-arch-type: cuda
85-
gpu-arch-version: "11.8"
86-
fail-fast: false
87-
uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
88-
with:
89-
repository: pytorch/vision
90-
runner: ${{ matrix.runner }}
91-
gpu-arch-type: ${{ matrix.gpu-arch-type }}
92-
gpu-arch-version: ${{ matrix.gpu-arch-version }}
93-
timeout: 120
94-
test-infra-ref: main
95-
script: |
96-
set -euxo pipefail
97-
98-
export PYTHON_VERSION=${{ matrix.python-version }}
99-
export VC_YEAR=2019
100-
export VSDEVCMD_ARGS=""
101-
export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }}
102-
export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }}
103-
104-
./.github/scripts/unittest.sh
105-
106-
onnx:
107-
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
108-
with:
109-
repository: pytorch/vision
110-
test-infra-ref: main
111-
script: |
112-
set -euo pipefail
113-
114-
export PYTHON_VERSION=3.10
115-
export GPU_ARCH_TYPE=cpu
116-
export GPU_ARCH_VERSION=''
117-
118-
./.github/scripts/setup-env.sh
119-
120-
# Prepare conda
121-
CONDA_PATH=$(which conda)
122-
eval "$(${CONDA_PATH} shell.bash hook)"
123-
conda activate ci
124-
125-
echo '::group::Install ONNX'
126-
pip install --progress-bar=off onnx onnxruntime
127-
echo '::endgroup::'
128-
129-
echo '::group::Install testing utilities'
130-
pip install --progress-bar=off pytest "numpy<2"
131-
echo '::endgroup::'
132-
133-
echo '::group::Run ONNX tests'
134-
pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_onnx.py
135-
echo '::endgroup::'
136-
137-
unittests-extended:
138-
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
139-
if: contains(github.event.pull_request.labels.*.name, 'run-extended')
140-
with:
141-
repository: pytorch/vision
142-
test-infra-ref: main
143-
script: |
144-
set -euo pipefail
145-
146-
export PYTHON_VERSION=3.9
147-
export GPU_ARCH_TYPE=cpu
148-
export GPU_ARCH_VERSION=''
149-
150-
./.github/scripts/setup-env.sh
151-
152-
# Prepare conda
153-
CONDA_PATH=$(which conda)
154-
eval "$(${CONDA_PATH} shell.bash hook)"
155-
conda activate ci
156-
157-
echo '::group::Pre-download model weights'
158-
pip install --progress-bar=off aiohttp aiofiles tqdm
159-
python scripts/download_model_urls.py
160-
echo '::endgroup::'
161-
162-
echo '::group::Install testing utilities'
163-
# TODO: remove the <8 constraint on pytest when https://github.com/pytorch/vision/issues/8238 is closed
164-
pip install --progress-bar=off "pytest<8"
165-
echo '::endgroup::'
166-
167-
echo '::group::Run extended unittests'
168-
export PYTORCH_TEST_WITH_EXTENDED=1
169-
pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_extended_*.py
170-
echo '::endgroup::'
46+
# unittests-macos:
47+
# strategy:
48+
# matrix:
49+
# python-version:
50+
# - "3.9"
51+
# - "3.10"
52+
# - "3.11"
53+
# - "3.12"
54+
# runner: ["macos-m1-stable"]
55+
# fail-fast: false
56+
# uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
57+
# with:
58+
# repository: pytorch/vision
59+
# timeout: 240
60+
# runner: ${{ matrix.runner }}
61+
# test-infra-ref: main
62+
# script: |
63+
# set -euo pipefail
64+
65+
# export PYTHON_VERSION=${{ matrix.python-version }}
66+
# export GPU_ARCH_TYPE=cpu
67+
# export GPU_ARCH_VERSION=''
68+
69+
# ${CONDA_RUN} ./.github/scripts/unittest.sh
70+
71+
# unittests-windows:
72+
# strategy:
73+
# matrix:
74+
# python-version:
75+
# - "3.9"
76+
# - "3.10"
77+
# - "3.11"
78+
# - "3.12"
79+
# runner: ["windows.4xlarge"]
80+
# gpu-arch-type: ["cpu"]
81+
# include:
82+
# - python-version: "3.9"
83+
# runner: windows.g5.4xlarge.nvidia.gpu
84+
# gpu-arch-type: cuda
85+
# gpu-arch-version: "11.8"
86+
# fail-fast: false
87+
# uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
88+
# with:
89+
# repository: pytorch/vision
90+
# runner: ${{ matrix.runner }}
91+
# gpu-arch-type: ${{ matrix.gpu-arch-type }}
92+
# gpu-arch-version: ${{ matrix.gpu-arch-version }}
93+
# timeout: 120
94+
# test-infra-ref: main
95+
# script: |
96+
# set -euxo pipefail
97+
98+
# export PYTHON_VERSION=${{ matrix.python-version }}
99+
# export VC_YEAR=2019
100+
# export VSDEVCMD_ARGS=""
101+
# export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }}
102+
# export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }}
103+
104+
# ./.github/scripts/unittest.sh
105+
106+
# onnx:
107+
# uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
108+
# with:
109+
# repository: pytorch/vision
110+
# test-infra-ref: main
111+
# script: |
112+
# set -euo pipefail
113+
114+
# export PYTHON_VERSION=3.10
115+
# export GPU_ARCH_TYPE=cpu
116+
# export GPU_ARCH_VERSION=''
117+
118+
# ./.github/scripts/setup-env.sh
119+
120+
# # Prepare conda
121+
# CONDA_PATH=$(which conda)
122+
# eval "$(${CONDA_PATH} shell.bash hook)"
123+
# conda activate ci
124+
125+
# echo '::group::Install ONNX'
126+
# pip install --progress-bar=off onnx onnxruntime
127+
# echo '::endgroup::'
128+
129+
# echo '::group::Install testing utilities'
130+
# pip install --progress-bar=off pytest "numpy<2"
131+
# echo '::endgroup::'
132+
133+
# echo '::group::Run ONNX tests'
134+
# pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_onnx.py
135+
# echo '::endgroup::'
136+
137+
# unittests-extended:
138+
# uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
139+
# if: contains(github.event.pull_request.labels.*.name, 'run-extended')
140+
# with:
141+
# repository: pytorch/vision
142+
# test-infra-ref: main
143+
# script: |
144+
# set -euo pipefail
145+
146+
# export PYTHON_VERSION=3.9
147+
# export GPU_ARCH_TYPE=cpu
148+
# export GPU_ARCH_VERSION=''
149+
150+
# ./.github/scripts/setup-env.sh
151+
152+
# # Prepare conda
153+
# CONDA_PATH=$(which conda)
154+
# eval "$(${CONDA_PATH} shell.bash hook)"
155+
# conda activate ci
156+
157+
# echo '::group::Pre-download model weights'
158+
# pip install --progress-bar=off aiohttp aiofiles tqdm
159+
# python scripts/download_model_urls.py
160+
# echo '::endgroup::'
161+
162+
# echo '::group::Install testing utilities'
163+
# # TODO: remove the <8 constraint on pytest when https://github.com/pytorch/vision/issues/8238 is closed
164+
# pip install --progress-bar=off "pytest<8"
165+
# echo '::endgroup::'
166+
167+
# echo '::group::Run extended unittests'
168+
# export PYTORCH_TEST_WITH_EXTENDED=1
169+
# pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_extended_*.py
170+
# echo '::endgroup::'

0 commit comments

Comments
 (0)