From 7d9116d9d7cb50aabd4d3748b64ac1462e120aab Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Tue, 30 Jun 2026 13:22:22 +0800 Subject: [PATCH] Fix test_type_based_dispatcher after TokenizedGenerateReqInput field changes (#29376) --- test/registered/utils/test_type_based_dispatcher.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/test/registered/utils/test_type_based_dispatcher.py b/test/registered/utils/test_type_based_dispatcher.py index 430e9ea4e..a4c29a71a 100644 --- a/test/registered/utils/test_type_based_dispatcher.py +++ b/test/registered/utils/test_type_based_dispatcher.py @@ -124,7 +124,9 @@ class TestTypeBasedDispatcher(unittest.TestCase): TokenizedGenerateReqInput( input_text="", input_ids=[1, 2], + input_embeds=None, mm_inputs=dict(), + token_type_ids=None, sampling_params=SamplingParams(), return_logprob=False, logprob_start_len=0, @@ -138,7 +140,7 @@ class TestTypeBasedDispatcher(unittest.TestCase): TokenizedEmbeddingReqInput( input_text="", input_ids=[1, 2], - image_inputs=dict(), + mm_inputs=dict(), token_type_ids=[1, 2], sampling_params=SamplingParams(), ) @@ -150,7 +152,9 @@ class TestTypeBasedDispatcher(unittest.TestCase): TokenizedGenerateReqInput( input_text="", input_ids=[1, 2], + input_embeds=None, mm_inputs=dict(), + token_type_ids=None, sampling_params=SamplingParams(), return_logprob=False, logprob_start_len=0, @@ -167,7 +171,7 @@ class TestTypeBasedDispatcher(unittest.TestCase): TokenizedEmbeddingReqInput( input_text="", input_ids=[1, 2], - image_inputs=dict(), + mm_inputs=dict(), token_type_ids=[1, 2], sampling_params=SamplingParams(), )