From 6a340c1fd3a656159689e45722e388a9e43d99eb Mon Sep 17 00:00:00 2001
From: Farzad Abdolhosseini <farzad@fixie.ai>
Date: Fri, 27 Sep 2024 16:42:39 -0700
Subject: [PATCH] rename ts to timestamp
---
ultravox/tools/ds_tool/ds_tool.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/ultravox/tools/ds_tool/ds_tool.py b/ultravox/tools/ds_tool/ds_tool.py
index 3a8d08bc..bb83245c 100644
--- 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,
)