From a40672b1d69bd6220b57c909ff4571e4698325ac Mon Sep 17 00:00:00 2001
From: Nicolas Hug <contact@nicolas-hug.com>
Date: Tue, 22 Apr 2025 14:44:53 +0100
Subject: [PATCH 1/3] Remove torchvision test dependency

---
 .github/workflows/linux_cuda_wheel.yaml    | 2 +-
 .github/workflows/linux_wheel.yaml         | 1 -
 .github/workflows/macos_wheel.yaml         | 1 -
 .github/workflows/reference_resources.yaml | 2 +-
 CONTRIBUTING.md                            | 2 +-
 5 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml
index a3abf1c5a..eaebef3f3 100644
--- a/.github/workflows/linux_cuda_wheel.yaml
+++ b/.github/workflows/linux_cuda_wheel.yaml
@@ -105,7 +105,7 @@ jobs:
         run: ${CONDA_RUN} python -m pip install --upgrade pip
       - name: Install PyTorch
         run: |
-          ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
+          ${CONDA_RUN} python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
           ${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
       - name: Install torchcodec from the wheel
         run: |
diff --git a/.github/workflows/linux_wheel.yaml b/.github/workflows/linux_wheel.yaml
index c20d348c9..1855e904e 100644
--- a/.github/workflows/linux_wheel.yaml
+++ b/.github/workflows/linux_wheel.yaml
@@ -104,7 +104,6 @@ jobs:
 
       - name: Install test dependencies
         run: |
-          python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           # Ideally we would find a way to get those dependencies from pyproject.toml
           python -m pip install numpy pytest pillow
 
diff --git a/.github/workflows/macos_wheel.yaml b/.github/workflows/macos_wheel.yaml
index 64448a0fa..ee436b7a4 100644
--- a/.github/workflows/macos_wheel.yaml
+++ b/.github/workflows/macos_wheel.yaml
@@ -103,7 +103,6 @@ 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 numpy pytest pillow
 
       - name: Delete the src/ folder just for fun
diff --git a/.github/workflows/reference_resources.yaml b/.github/workflows/reference_resources.yaml
index c3a10fef9..7471134ae 100644
--- a/.github/workflows/reference_resources.yaml
+++ b/.github/workflows/reference_resources.yaml
@@ -42,7 +42,7 @@ jobs:
         run: |
           # Note that we're installing stable - this is for running a script where we're a normal PyTorch
           # user, not for building TorhCodec.
-          python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
           python -m pip install numpy pillow
 
       - name: Check out repo
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
index 278090b9f..af19028f3 100644
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -48,7 +48,7 @@ pip install -e ".[dev]" --no-build-isolation -vv
 
 ### Running unit tests
 
-To run python tests run (please make sure `torchvision` is installed):
+To run python tests run:
 
 ```bash
 pytest test -vvv

From 9a4304988eab681500bcf29695dfac47ee54cc4e Mon Sep 17 00:00:00 2001
From: Nicolas Hug <contact@nicolas-hug.com>
Date: Tue, 22 Apr 2025 15:25:23 +0100
Subject: [PATCH 2/3] Put back dep for CUDA job since we run the benchmarks

---
 .github/workflows/linux_cuda_wheel.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml
index eaebef3f3..a3abf1c5a 100644
--- a/.github/workflows/linux_cuda_wheel.yaml
+++ b/.github/workflows/linux_cuda_wheel.yaml
@@ -105,7 +105,7 @@ jobs:
         run: ${CONDA_RUN} python -m pip install --upgrade pip
       - name: Install PyTorch
         run: |
-          ${CONDA_RUN} python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
+          ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
           ${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
       - name: Install torchcodec from the wheel
         run: |

From 2dae6ab7455371b89e0b71802442de6608a5f0f7 Mon Sep 17 00:00:00 2001
From: Nicolas Hug <contact@nicolas-hug.com>
Date: Tue, 22 Apr 2025 17:04:21 +0100
Subject: [PATCH 3/3] empty