[Bugfix] Fix arguments passed to Sequence in stop checker test (vll…

…m-project#5092)
neuralmagic · Jul 14, 2024 · c420496 · c420496
1 parent bf29f34
commit c420496
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,
     )