lint

apache · Sep 23, 2024 · 93bbd49 · 93bbd49
1 parent 7703f53
commit 93bbd49
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/ml/inference/vllm_inference.py b/sdks/python/apache_beam/ml/inference/vllm_inference.py
@@ -100,7 +100,7 @@ def __init__(self, model_name: str, vllm_server_kwargs: Dict[str, str]):
     self._vllm_server_kwargs = vllm_server_kwargs
     self._server_started = False
     self._server_process = None
-    self._server_port = None
+    self._server_port: int = -1
 
     self.start_server()
 
@@ -218,7 +218,7 @@ def __init__(
       self,
       model_name: str,
       chat_template_path: Optional[str] = None,
-      vllm_server_kwargs: Dict[str, str] = None):
+      vllm_server_kwargs: Optional[Dict[str, str]] = None):
     """ Implementation of the ModelHandler interface for vLLM using previous
     messages as input.