diff --git a/test/registered/unit/managers/test_scheduler_init_req_max_new_tokens.py b/test/registered/unit/managers/test_scheduler_init_req_max_new_tokens.py index 7e4921c27..bfe542dfd 100644 --- a/test/registered/unit/managers/test_scheduler_init_req_max_new_tokens.py +++ b/test/registered/unit/managers/test_scheduler_init_req_max_new_tokens.py @@ -46,6 +46,7 @@ class TestSchedulerInitReqMaxNewTokens(unittest.TestCase): scheduler.max_req_len = max_req_len scheduler.max_total_num_tokens = max_total_num_tokens scheduler.page_size = page_size + scheduler.server_args = SimpleNamespace(dcp_size=1) scheduler.max_new_tokens_limit = envs.SGLANG_MAX_NEW_TOKENS_LIMIT.get() return scheduler