Fix: AudioCaps n_channels argument.

Labbeti · Jan 2, 2024 · 2e4c1b9 · 2e4c1b9
1 parent 1247035
commit 2e4c1b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aac_datasets/datasets/functional/audiocaps.py b/src/aac_datasets/datasets/functional/audiocaps.py
@@ -442,9 +442,9 @@ def download_audiocaps_datasets(
     # AudioCaps-specific args
     audio_duration: float = 10.0,
     audio_format: str = "flac",
+    audio_n_channels: int = 1,
     download_audio: bool = True,
     ffmpeg_path: Union[str, Path, None] = None,
-    n_channels: int = 1,
     sr: int = 32_000,
     verify_files: bool = False,
     ytdlp_path: Union[str, Path, None] = None,
@@ -462,9 +462,9 @@ def download_audiocaps_datasets(
         verbose=verbose,
         audio_duration=audio_duration,
         audio_format=audio_format,
+        audio_n_channels=audio_n_channels,
         download_audio=download_audio,
         ffmpeg_path=ffmpeg_path,
-        n_channels=n_channels,
         sr=sr,
         verify_files=verify_files,
         ytdlp_path=ytdlp_path,