From 5f283991e9983dcc367317e5d13e2546b7f1e4e4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabian=20Prei=C3=9F?= Date: Wed, 12 Jun 2024 01:37:27 +0200 Subject: [PATCH] [Minor] Correct Optional type hints in api (#526) --- python/sglang/api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sglang/api.py b/python/sglang/api.py index ff113ab9b..d21024520 100644 --- a/python/sglang/api.py +++ b/python/sglang/api.py @@ -43,14 +43,14 @@ def set_default_backend(backend: BaseBackend): global_config.default_backend = backend -def flush_cache(backend: BaseBackend = None): +def flush_cache(backend: Optional[BaseBackend] = None): backend = backend or global_config.default_backend if backend is None: return False return backend.flush_cache() -def get_server_args(backend: BaseBackend = None): +def get_server_args(backend: Optional[BaseBackend] = None): backend = backend or global_config.default_backend if backend is None: return None @@ -158,7 +158,7 @@ def video(path: str, num_frames: int): def select( name: Optional[str] = None, - choices: List[str] = None, + choices: Optional[List[str]] = None, temperature: float = 0.0, ): assert choices is not None