From c7424c1b3df22b12da9add6e0ce14941d6943f75 Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Wed, 29 May 2024 01:53:27 +0000 Subject: [PATCH] Fix arguments passed to `Sequence` --- tests/engine/output_processor/test_stop_checker.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/engine/output_processor/test_stop_checker.py b/tests/engine/output_processor/test_stop_checker.py index ae54c83605e1..1d9c878ddde5 100644 --- a/tests/engine/output_processor/test_stop_checker.py +++ b/tests/engine/output_processor/test_stop_checker.py @@ -15,8 +15,11 @@ def sequence_with_eos(text: str, eos_token: str, """ seq = Sequence( seq_id=0, - prompt="", - prompt_token_ids=[], + inputs={ + "prompt": "", + "prompt_token_ids": [], + "multi_modal_data": None, + }, block_size=16, eos_token_id=eos_token_id, )