Fix MockTokenizer in the unit tests (#1774)

87a7cfa0 · Lianmin Zheng · GitHub · 8f8f96a6 · 87a7cfa0
Unverified Commit 87a7cfa0 authored Oct 23, 2024 by Lianmin Zheng Committed by GitHub Oct 23, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

python/sglang/test/srt/sampling/penaltylib/utils.py python/sglang/test/srt/sampling/penaltylib/utils.py +1 -0

No files found.
--- a/python/sglang/test/srt/sampling/penaltylib/utils.py
+++ b/python/sglang/test/srt/sampling/penaltylib/utils.py
@@ -24,6 +24,7 @@ class MockSamplingParams:
 @dataclasses.dataclass
 class MockTokenizer:
    eos_token_id: int
+    additional_stop_token_ids: typing.Optional[typing.List[int]] = None


 @dataclasses.dataclass