diff --git a/.github/workflows/wheel.yaml b/.github/workflows/wheel.yaml index 81e8df2a..1e11e6b2 100644 --- a/.github/workflows/wheel.yaml +++ b/.github/workflows/wheel.yaml @@ -58,7 +58,7 @@ jobs: strategy: fail-fast: false matrix: - python-version: ['3.9'] + python-version: ['3.9', '3.10', '3.11', '3.12'] ffmpeg-version-for-tests: ['4.4.2', '5.1.2', '6.1.1', '7.0.1'] if: ${{ always() }} needs: build @@ -78,7 +78,7 @@ jobs: run: python -m pip install --upgrade pip - name: Install PyTorch run: | - python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu + python -m pip install torch --index-url https://download.pytorch.org/whl/cpu - name: Install torchcodec from the wheel run: | wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"` @@ -101,9 +101,10 @@ jobs: - name: Install test dependencies run: | - python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu + python -m pip install torchvision --index-url https://download.pytorch.org/whl/cpu # Ideally we would find a way to get those dependencies from pyproject.toml python -m pip install numpy pytest pillow + python -c "import torch; print(f'{torch.__version__ = }')" - name: Delete the src/ folder just for fun run: | diff --git a/examples/basic_example.py b/examples/basic_example.py index 6f556711..abbc1b46 100644 --- a/examples/basic_example.py +++ b/examples/basic_example.py @@ -171,4 +171,3 @@ def plot(frames: torch.Tensor, title : Optional[str] = None): # %% plot(frame_at_2_seconds.data, "Frame displayed at 2 seconds") plot(first_two_seconds.data, "Frames displayed during [0, 2) seconds") - diff --git a/src/torchcodec/__init__.py b/src/torchcodec/__init__.py index a97c32ce..79957dd1 100644 --- a/src/torchcodec/__init__.py +++ b/src/torchcodec/__init__.py @@ -6,4 +6,4 @@ from . import decoders # noqa -__version__ = "0.0.2.dev" +__version__ = "0.0.3"