diff --git a/tests/core/utils.py b/tests/core/utils.py index bff015da91e86..f249f4b59a2ee 100644 --- 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