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

18c1f16d · Cyrus Leung · GitHub · 5bd3c650 · 18c1f16d
Unverified Commit 18c1f16d authored May 29, 2024 by Cyrus Leung Committed by GitHub May 29, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

tests/engine/output_processor/test_stop_checker.py tests/engine/output_processor/test_stop_checker.py +5 -2

No files found.
--- 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="",
+        inputs={
-        prompt_token_ids=[],
+            "prompt": "",
+            "prompt_token_ids": [],
+            "multi_modal_data": None,
+        },
        block_size=16,
        eos_token_id=eos_token_id,
    )