Skip to content

Commit 89fe2ea

Browse files
NicolasHugNicolasHug
authored
0.2.0 Release (#491)
1 parent 93fff37 commit 89fe2ea

8 files changed

+31
-29
lines changed

‎.github/workflows/cpp_tests.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ jobs:
3434
run: python -m pip install --upgrade pip
3535
- name: Install dependencies
3636
run: |
37-
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
37+
python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
3838
- name: Install ffmpeg and pkg-config
3939
run: |
4040
conda install "ffmpeg=${{ matrix.ffmpeg-version-for-tests }}" pkg-config -c conda-forge

‎.github/workflows/docs.yaml

+5-5
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,12 @@ defaults:
1515

1616
jobs:
1717
generate-matrix:
18-
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
18+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.6-torchcodec
1919
with:
2020
package-type: wheel
2121
os: linux
2222
test-infra-repository: pytorch/test-infra
23-
test-infra-ref: main
23+
test-infra-ref: release/2.6-torchcodec
2424
with-cpu: disable
2525
with-xpu: disable
2626
with-rocm: disable
@@ -31,12 +31,12 @@ jobs:
3131
strategy:
3232
fail-fast: false
3333
name: Build and Upload wheel
34-
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
34+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.6-torchcodec
3535
with:
3636
repository: pytorch/torchcodec
3737
ref: ""
3838
test-infra-repository: pytorch/test-infra
39-
test-infra-ref: main
39+
test-infra-ref: release/2.6-torchcodec
4040
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
4141
post-script: packaging/post_build_script.sh
4242
smoke-test-script: packaging/fake_smoke_test.py
@@ -93,7 +93,7 @@ jobs:
9393
run: ${CONDA_RUN} python -m pip install --upgrade pip
9494
- name: Install PyTorch
9595
run: |
96-
${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
96+
${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cu${{ env.cuda_version_without_periods }}
9797
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
9898
- name: Install torchcodec from the wheel
9999
run: |

‎.github/workflows/lint.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ jobs:
6262
run: python -m pip install --upgrade pip
6363
- name: Install dependencies and FFmpeg
6464
run: |
65-
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
65+
python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
6666
conda install "ffmpeg=7.0.1" pkg-config -c conda-forge
6767
ffmpeg -version
6868
- name: Build and install torchcodec

‎.github/workflows/linux_cuda_wheel.yaml

+7-7
Original file line numberDiff line numberDiff line change
@@ -25,12 +25,12 @@ defaults:
2525

2626
jobs:
2727
generate-matrix:
28-
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
28+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.6-torchcodec
2929
with:
3030
package-type: wheel
3131
os: linux
3232
test-infra-repository: pytorch/test-infra
33-
test-infra-ref: main
33+
test-infra-ref: release/2.6-torchcodec
3434
with-cpu: disable
3535
with-xpu: disable
3636
with-rocm: disable
@@ -41,12 +41,12 @@ jobs:
4141
strategy:
4242
fail-fast: false
4343
name: Build and Upload wheel
44-
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
44+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.6-torchcodec
4545
with:
4646
repository: pytorch/torchcodec
4747
ref: ""
4848
test-infra-repository: pytorch/test-infra
49-
test-infra-ref: main
49+
test-infra-ref: release/2.6-torchcodec
5050
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
5151
post-script: packaging/post_build_script.sh
5252
smoke-test-script: packaging/fake_smoke_test.py
@@ -65,10 +65,10 @@ jobs:
6565
# PR.
6666
# For the actual release we should add that label and change this to
6767
# include more python versions.
68-
python-version: ['3.9']
68+
python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
6969
cuda-version: ['11.8', '12.4', '12.6']
7070
# TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
71-
ffmpeg-version-for-tests: ['6', '7']
71+
ffmpeg-version-for-tests: ['5', '6', '7']
7272
container:
7373
image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}"
7474
options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility"
@@ -106,7 +106,7 @@ jobs:
106106
run: ${CONDA_RUN} python -m pip install --upgrade pip
107107
- name: Install PyTorch
108108
run: |
109-
${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
109+
${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cu${{ env.cuda_version_without_periods }}
110110
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
111111
- name: Install torchcodec from the wheel
112112
run: |

‎.github/workflows/linux_wheel.yaml

+7-7
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,12 @@ defaults:
2626
jobs:
2727

2828
generate-matrix:
29-
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
29+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.6-torchcodec
3030
with:
3131
package-type: wheel
3232
os: linux
3333
test-infra-repository: pytorch/test-infra
34-
test-infra-ref: main
34+
test-infra-ref: release/2.6-torchcodec
3535
with-xpu: disable
3636
with-rocm: disable
3737
with-cuda: disable
@@ -42,12 +42,12 @@ jobs:
4242
strategy:
4343
fail-fast: false
4444
name: Build and Upload Linux wheel
45-
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
45+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.6-torchcodec
4646
with:
4747
repository: pytorch/torchcodec
4848
ref: ""
4949
test-infra-repository: pytorch/test-infra
50-
test-infra-ref: main
50+
test-infra-ref: release/2.6-torchcodec
5151
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
5252
post-script: packaging/post_build_script.sh
5353
smoke-test-script: packaging/fake_smoke_test.py
@@ -61,7 +61,7 @@ jobs:
6161
strategy:
6262
fail-fast: false
6363
matrix:
64-
python-version: ['3.9']
64+
python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
6565
ffmpeg-version-for-tests: ['4.4.2', '5.1.2', '6.1.1', '7.0.1']
6666
needs: build
6767
steps:
@@ -80,7 +80,7 @@ jobs:
8080
run: python -m pip install --upgrade pip
8181
- name: Install PyTorch
8282
run: |
83-
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
83+
python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
8484
- name: Install torchcodec from the wheel
8585
run: |
8686
wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"`
@@ -103,7 +103,7 @@ jobs:
103103
104104
- name: Install test dependencies
105105
run: |
106-
python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
106+
python -m pip install torchvision --index-url https://download.pytorch.org/whl/cpu
107107
# Ideally we would find a way to get those dependencies from pyproject.toml
108108
python -m pip install numpy pytest pillow
109109

‎.github/workflows/macos_wheel.yaml

+7-7
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,12 @@ defaults:
2626
jobs:
2727

2828
generate-matrix:
29-
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
29+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.6-torchcodec
3030
with:
3131
package-type: wheel
3232
os: macos-arm64
3333
test-infra-repository: pytorch/test-infra
34-
test-infra-ref: main
34+
test-infra-ref: release/2.6-torchcodec
3535
with-xpu: disable
3636
with-rocm: disable
3737
with-cuda: disable
@@ -42,12 +42,12 @@ jobs:
4242
strategy:
4343
fail-fast: false
4444
name: Build and Upload Mac wheel
45-
uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@main
45+
uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@release/2.6-torchcodec
4646
with:
4747
repository: pytorch/torchcodec
4848
ref: ""
4949
test-infra-repository: pytorch/test-infra
50-
test-infra-ref: main
50+
test-infra-ref: release/2.6-torchcodec
5151
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
5252
post-script: packaging/post_build_script.sh
5353
smoke-test-script: packaging/fake_smoke_test.py
@@ -62,7 +62,7 @@ jobs:
6262
strategy:
6363
fail-fast: false
6464
matrix:
65-
python-version: ['3.9']
65+
python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
6666
ffmpeg-version-for-tests: ['4.4.2', '5.1.2', '6.1.1', '7.0.1']
6767
needs: build
6868
steps:
@@ -84,7 +84,7 @@ jobs:
8484

8585
- name: Install PyTorch
8686
run: |
87-
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
87+
python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
8888
8989
- name: Install torchcodec from the wheel
9090
run: |
@@ -102,7 +102,7 @@ jobs:
102102
103103
- name: Install test dependencies
104104
run: |
105-
python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
105+
python -m pip install torchvision --index-url https://download.pytorch.org/whl/cpu
106106
python -m pip install numpy pytest pillow
107107
108108
- name: Delete the src/ folder just for fun

‎setup.py

+2
Original file line numberDiff line numberDiff line change
@@ -178,12 +178,14 @@ def _write_version_files():
178178
# the content of `version.txt` plus some suffix like "+cpu" or "+cu112".
179179
# See
180180
# https://github.com/pytorch/test-infra/blob/61e6da7a6557152eb9879e461a26ad667c15f0fd/tools/pkg-helpers/pytorch_pkg_helpers/version.py#L113
181+
version = version.replace("+cpu", "")
181182
with open(_ROOT_DIR / "version.txt", "w") as f:
182183
f.write(f"{version}")
183184
else:
184185
with open(_ROOT_DIR / "version.txt") as f:
185186
version = f.readline().strip()
186187
try:
188+
version = version.replace("+cpu", "")
187189
sha = (
188190
subprocess.check_output(
189191
["git", "rev-parse", "HEAD"], cwd=str(_ROOT_DIR)

‎version.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0.1.2a0
1+
0.2.0

0 commit comments

Comments
 (0)