fix(tgi): typer does not support Union types (#219)

huggingface · Sep 8, 2023 · 5b531fd · 5b531fd
1 parent 6fbd1c5
commit 5b531fd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/text-generation-inference/server/text_generation_server/cli.py b/text-generation-inference/server/text_generation_server/cli.py
@@ -1,6 +1,5 @@
 import sys
-from pathlib import Path
-from typing import Optional, Union
+from typing import Optional
 
 import typer
 from loguru import logger
@@ -15,7 +14,7 @@ def serve(
     revision: Optional[str] = None,
     sharded: bool = False,
     trust_remote_code: bool = None,
-    uds_path: Union[Path, str] = "/tmp/text-generation-server",
+    uds_path: str = "/tmp/text-generation-server",
     logger_level: str = "INFO",
     json_output: bool = False,
 ):