Fix MockTokenizer in the unit tests (#1774)

sgl-project · Oct 24, 2024 · 87a7cfa · 87a7cfa
1 parent 8f8f96a
commit 87a7cfa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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