From a47187ebbaca2a1c500d1bc1a093cb4ee9789f21 Mon Sep 17 00:00:00 2001
From: hippi345 <13539685+hippi345@users.noreply.github.com>
Date: Thu, 15 Oct 2026 04:07:44 +0000
Subject: [PATCH] Replace deprecated torchvision decode_image with TorchCodec

The torchvision.io.decode_image function is deprecated in favor of
TorchCodec. This change updates beginner_source/basics/data_tutorial.py
to use torchcodec.decoders.decode_image instead.

The change maintains API compatibility as TorchCodec's decode_image
accepts the same arguments (file path) and returns tensors in the same
format (CHW uint8 by default).

Also adds torchcodec to requirements.txt to ensure it is available
during builds.

Fixes #3978
---
 .ci/docker/requirements.txt             | 1 +
 beginner_source/basics/data_tutorial.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
index 58cbcb304d2..daf515b6a7d 100644
--- a/.ci/docker/requirements.txt
+++ b/.ci/docker/requirements.txt
@@ -31,6 +31,7 @@ librosa
 torch==2.14
 cuda-bindings>=13.1.0  # Required for CUDA graph annotations tutorial
 torchvision
+torchcodec
 torchdata
 networkx
 PyHamcrest
diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py
index c49f25a587a..937768a311e 100644
--- a/beginner_source/basics/data_tutorial.py
+++ b/beginner_source/basics/data_tutorial.py
@@ -120,7 +120,7 @@
 
 import os
 import pandas as pd
-from torchvision.io import decode_image
+from torchcodec.decoders import decode_image
 
 class CustomImageDataset(Dataset):
     def __init__(self, annotations_file, img_dir, transform=None, target_transform=None):