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'):