Mod: Update AudioMetaData import again.

Labbeti · Jan 5, 2024 · a7f3786 · a7f3786
1 parent fb8912c
commit a7f3786
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/aac_datasets/datasets/audiocaps.py b/src/aac_datasets/datasets/audiocaps.py
@@ -22,10 +22,10 @@
 from typing_extensions import TypedDict, NotRequired
 
 try:
-    # To support torchaudio < 2.1.0
-    from torchaudio.backend.common import AudioMetaData
-except (ImportError, ModuleNotFoundError):
+    # 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
 from aac_datasets.datasets.functional.audiocaps import (

diff --git a/src/aac_datasets/datasets/base.py b/src/aac_datasets/datasets/base.py
@@ -24,10 +24,10 @@
 from typing_extensions import TypedDict
 
 try:
-    # To support torchaudio < 2.1.0
-    from torchaudio.backend.common import AudioMetaData
-except (ImportError, ModuleNotFoundError):
+    # 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