Fix MockTokenizer in the unit tests (#1774)

2024-10-23 17:47:05 -07:00
parent 8f8f96a621
commit 87a7cfa080
1 changed files with 1 additions and 0 deletions
--- 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