Mod: Update AudioMetaData import for torchaudio > 2.1.0.

Labbeti · Oct 5, 2023 · 7b73536 · 7b73536
1 parent 8d1c71a
commit 7b73536
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/aac_datasets/datasets/audiocaps.py b/src/aac_datasets/datasets/audiocaps.py
@@ -25,9 +25,14 @@
 
 from torch import Tensor
 from torch.hub import download_url_to_file
-from torchaudio.backend.common import AudioMetaData
 from typing_extensions import TypedDict, NotRequired
 
+try:
+    # To support torchaudio >=2.1.0
+    from torchaudio import AudioMetaData  # type: ignore
+except ImportError:
+    from torchaudio.backend.common import AudioMetaData
+
 from aac_datasets.datasets.base import AACDataset, DatasetCard
 from aac_datasets.utils.paths import _get_root, _get_ffmpeg_path, _get_ytdl_path
 

diff --git a/src/aac_datasets/datasets/base.py b/src/aac_datasets/datasets/base.py
@@ -21,9 +21,14 @@
 import torchaudio
 import tqdm
 
-from torchaudio.backend.common import AudioMetaData
 from typing_extensions import TypedDict
 
+try:
+    # To support torchaudio >=2.1.0
+    from torchaudio import AudioMetaData  # type: ignore
+except ImportError:
+    from torchaudio.backend.common import AudioMetaData
+
 from torch import Tensor
 from torch.utils.data.dataset import Dataset