From aa19691fb007afed0f84abaa191120aa7d963d62 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 16 Jul 2025 10:18:18 +0100 Subject: [PATCH 1/2] Let load and save rely on *_with_torchcodec --- src/torchaudio/__init__.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/torchaudio/__init__.py b/src/torchaudio/__init__.py index e533cafe9d..1fde90b871 100644 --- a/src/torchaudio/__init__.py +++ b/src/torchaudio/__init__.py @@ -7,8 +7,6 @@ get_audio_backend as _get_audio_backend, info as _info, list_audio_backends as _list_audio_backends, - load, - save, set_audio_backend as _set_audio_backend, ) from ._torchcodec import load_with_torchcodec, save_with_torchcodec @@ -41,6 +39,13 @@ pass +def load(*args, **kwargs): + return load_with_torchcodec(*args, **kwargs) + +def save(*args, **kwargs): + return save_with_torchcodec(*args, **kwargs) + + __all__ = [ "AudioMetaData", "load", From c85011b1539c988809b1221f16f64e8a0621a3ae Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 16 Jul 2025 13:49:53 +0100 Subject: [PATCH 2/2] install torchcodec in doc job --- .github/workflows/build_docs.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build_docs.yml b/.github/workflows/build_docs.yml index e92c556218..f681e3b7ec 100644 --- a/.github/workflows/build_docs.yml +++ b/.github/workflows/build_docs.yml @@ -68,7 +68,7 @@ jobs: GPU_ARCH_ID=cu126 # This is hard-coded and must be consistent with gpu-arch-version. PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}" - pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}" + pip install --progress-bar=off --pre torch torchcodec --index-url="${PYTORCH_WHEEL_INDEX}" echo "::endgroup::" echo "::group::Install TorchAudio"