add typing

Author: LunrEclipse

vllm/utils.py

@@ -1414,7 +1414,7 @@ def get_beam_search_score(
     return cumulative_logprob / (seq_len**length_penalty)
 
 
-def create_sort_beams_key_function(eos_token_id: int, length_penalty):
+def create_sort_beams_key_function(eos_token_id: int, length_penalty: float):
 
     def sort_beams_key(x: BeamSearchSequence) -> float:
         return get_beam_search_score(x.tokens, x.cum_logprob, eos_token_id,