Fix incorrect return type annotation

vllm-project · Jun 3, 2024 · 71ace6c · 71ace6c
1 parent 5ef7804
commit 71ace6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/utils.py b/tests/core/utils.py
@@ -48,7 +48,7 @@ def create_dummy_prompt_encoder_decoder(
     lora_request: Optional[LoRARequest] = None,
     use_beam_search: bool = False,
     best_of: int = 1,
-) -> Tuple[Sequence, SequenceGroup]:
+) -> Tuple[Sequence, Sequence, SequenceGroup]:
     if not block_size:
         block_size = decoder_prompt_length