From b13778fefd71c955eb93bd9c4d3bb9c850acd5db Mon Sep 17 00:00:00 2001
From: Max Bain <36994049+m-bain@users.noreply.github.com>
Date: Wed, 7 Jun 2023 11:47:49 +0100
Subject: [PATCH] make aud optional

---
 whisperx/utils.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/whisperx/utils.py b/whisperx/utils.py
index ff17cce..36c7543 100644
--- a/whisperx/utils.py
+++ b/whisperx/utils.py
@@ -399,13 +399,15 @@ def get_writer(
     output_format: str, output_dir: str
 ) -> Callable[[dict, TextIO, dict], None]:
     writers = {
-        "aud": WriteAudacity,
         "txt": WriteTXT,
         "vtt": WriteVTT,
         "srt": WriteSRT,
         "tsv": WriteTSV,
         "json": WriteJSON,
     }
+    optional_writers = {
+        "aud": WriteAudacity,
+    }
 
     if output_format == "all":
         all_writers = [writer(output_dir) for writer in writers.values()]
@@ -416,6 +418,8 @@ def get_writer(
 
         return write_all
 
+    if output_format in optional_writers:
+        return optional_writers[output_format](output_dir)
     return writers[output_format](output_dir)
 
 def interpolate_nans(x, method='nearest'):