[Fix CI] Fix test_mamba_unittest.py (#16810)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
2026-01-10 07:03:47 +08:00
parent a6c29d4cbd
commit 08636f72b5
1 changed files with 4 additions and 1 deletions
--- a/test/registered/attention/test_mamba_unittest.py
+++ b/test/registered/attention/test_mamba_unittest.py
@@ -20,7 +20,10 @@ register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu")
 class TestMamba(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
-        pass
+        server_args = ServerArgs(model_path="dummy")
+        server_args.enable_dp_attention = False
+        server_args.page_size = 1
+        set_global_server_args_for_scheduler(server_args)

    @classmethod
    def tearDownClass(cls):