[Bugfix] Fix arguments passed to Sequence in stop checker test (#5092)

vllm-project · May 29, 2024 · 18c1f16 · 18c1f16
1 parent 5bd3c65
commit 18c1f16
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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,
     )