minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Fix test_type_based_dispatcher after TokenizedGenerateReqInput field changes (#29376)

Browse Source
This commit is contained in:
YC Yen-Ching Tseng
2026-06-29 22:22:22 -07:00
committed by GitHub
parent 7da4e30d9b
commit 7d9116d9d7
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/utils/test_type_based_dispatcher.py
+6 -2
View File
@@ -124,7 +124,9 @@ class TestTypeBasedDispatcher(unittest.TestCase):
TokenizedGenerateReqInput( TokenizedGenerateReqInput(
input_text="", input_text="",
input_ids=[1, 2], input_ids=[1, 2],
input_embeds=None,
mm_inputs=dict(), mm_inputs=dict(),
token_type_ids=None,
sampling_params=SamplingParams(), sampling_params=SamplingParams(),
return_logprob=False, return_logprob=False,
logprob_start_len=0, logprob_start_len=0,
@@ -138,7 +140,7 @@ class TestTypeBasedDispatcher(unittest.TestCase):
TokenizedEmbeddingReqInput( TokenizedEmbeddingReqInput(
input_text="", input_text="",
input_ids=[1, 2], input_ids=[1, 2],
image_inputs=dict(), mm_inputs=dict(),
token_type_ids=[1, 2], token_type_ids=[1, 2],
sampling_params=SamplingParams(), sampling_params=SamplingParams(),
) )
@@ -150,7 +152,9 @@ class TestTypeBasedDispatcher(unittest.TestCase):
TokenizedGenerateReqInput( TokenizedGenerateReqInput(
input_text="", input_text="",
input_ids=[1, 2], input_ids=[1, 2],
input_embeds=None,
mm_inputs=dict(), mm_inputs=dict(),
token_type_ids=None,
sampling_params=SamplingParams(), sampling_params=SamplingParams(),
return_logprob=False, return_logprob=False,
logprob_start_len=0, logprob_start_len=0,
@@ -167,7 +171,7 @@ class TestTypeBasedDispatcher(unittest.TestCase):
TokenizedEmbeddingReqInput( TokenizedEmbeddingReqInput(
input_text="", input_text="",
input_ids=[1, 2], input_ids=[1, 2],
image_inputs=dict(), mm_inputs=dict(),
token_type_ids=[1, 2], token_type_ids=[1, 2],
sampling_params=SamplingParams(), sampling_params=SamplingParams(),
) )