Add missing arguments to docstring (facebookresearch#254)

Dinglet · Aug 28, 2023 · 3901c1a · 3901c1a
1 parent f339be6
commit 3901c1a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/data/audio.py b/data/audio.py
@@ -162,6 +162,8 @@ def audio_write(stem_name: tp.Union[str, Path],
 
     Args:
         stem_name (str or Path): Filename without extension which will be added automatically.
+        wav (torch.Tensor): Audio data to save.
+        sample_rate (int): Sample rate of audio data.
         format (str): Either "wav" or "mp3".
         mp3_rate (int): kbps when using mp3s.
         normalize (bool): if `True` (default), normalizes according to the prescribed