Update deprecated Python 3.8 typing (#13971)

vllm/beam_search.py

@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Optional, Union
 
 from vllm.sequence import Logprob
 
@@ -17,14 +17,14 @@ class BeamSearchSequence:
     about to be returned to the user.
     """
     # The tokens includes the prompt.
-    tokens: List[int]
-    logprobs: List[Dict[int, Logprob]]
+    tokens: list[int]
+    logprobs: list[dict[int, Logprob]]
     cum_logprob: float = 0.0
     text: Optional[str] = None
     finish_reason: Optional[str] = None
     stop_reason: Union[int, str, None] = None
     multi_modal_data: Optional["MultiModalDataDict"] = None
-    mm_processor_kwargs: Optional[Dict[str, Any]] = None
+    mm_processor_kwargs: Optional[dict[str, Any]] = None
 
 
 @dataclass
@@ -33,20 +33,20 @@ class BeamSearchOutput:
     It contains the list of the best beam search sequences.
     The length of the list is equal to the beam width.
     """
-    sequences: List[BeamSearchSequence]
+    sequences: list[BeamSearchSequence]
 
 
 class BeamSearchInstance:
 
-    def __init__(self, prompt_tokens: List[int]):
-        self.beams: List[BeamSearchSequence] = [
+    def __init__(self, prompt_tokens: list[int]):
+        self.beams: list[BeamSearchSequence] = [
             BeamSearchSequence(tokens=prompt_tokens, logprobs=[])
         ]
-        self.completed: List[BeamSearchSequence] = []
+        self.completed: list[BeamSearchSequence] = []
 
 
 def get_beam_search_score(
-    tokens: List[int],
+    tokens: list[int],
     cumulative_logprob: float,
     eos_token_id: int,
     length_penalty: float = 1.0,