rename ts to timestamp

fixie-ai · Sep 27, 2024 · 6a340c1 · 6a340c1
1 parent 7b132c8
commit 6a340c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ultravox/tools/ds_tool/ds_tool.py b/ultravox/tools/ds_tool/ds_tool.py
@@ -396,7 +396,7 @@ class DatasetToolArgs:
 
     task: Union[TtsTask, TextGenerationTask, TimestampGenerationTask] = (
         simple_parsing.subgroups(
-            {"tts": TtsTask, "textgen": TextGenerationTask, "ts": TimestampGenerationTask},  # type: ignore
+            {"tts": TtsTask, "textgen": TextGenerationTask, "timestamp": TimestampGenerationTask},  # type: ignore
             default_factory=TtsTask,
             positional=True,
         )