From 87a7cfa080cec3f123618c1429b5f998bf5d99cb Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Wed, 23 Oct 2024 17:47:05 -0700 Subject: [PATCH] Fix MockTokenizer in the unit tests (#1774) --- python/sglang/test/srt/sampling/penaltylib/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/test/srt/sampling/penaltylib/utils.py b/python/sglang/test/srt/sampling/penaltylib/utils.py index 31667c7f0..4acfa5a41 100644 --- 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