Mod: Update internal typing.

Labbeti · Nov 21, 2023 · 1b772b5 · 1b772b5
1 parent 9584efb
commit 1b772b5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/aac_datasets/datasets/base.py b/src/aac_datasets/datasets/base.py
@@ -427,7 +427,8 @@ def _load_auto_value(self, column: str, idx: int) -> Any:
 
     def _load_audio(self, idx: int) -> Tensor:
         fpath = self.at(idx, "fpath")
-        audio, sr = torchaudio.load(fpath)  # type: ignore
+        audio_and_sr: Tuple[Tensor, int] = torchaudio.load(fpath)  # type: ignore
+        audio, sr = audio_and_sr
 
         # Sanity check
         if audio.nelement() == 0: