From 7cee8522e598c97fdbeda5f9d48354ddf04d80e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Carl=20Thom=C3=A9?= <carlthome@gmail.com>
Date: Mon, 28 Aug 2023 14:19:38 +0200
Subject: [PATCH] Add missing arguments to docstring (#254)

---
 audiocraft/data/audio.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/audiocraft/data/audio.py b/audiocraft/data/audio.py
index 39c87047..2ac5e6cf 100644
--- a/audiocraft/data/audio.py
+++ b/audiocraft/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