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 c9029f186..7e4921c27 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 @@ -6,7 +6,7 @@ from sglang.srt.environ import envs from sglang.srt.managers.scheduler import Scheduler from sglang.test.ci.ci_register import register_cpu_ci -register_cpu_ci(est_time=1, suite="stage-a-test-cpu") +register_cpu_ci(est_time=1, suite="base-a-test-cpu") class TestSchedulerInitReqMaxNewTokens(unittest.TestCase):