From 08636f72b5d9742673cf19b2dfea4880ee9581c7 Mon Sep 17 00:00:00 2001
From: Yuan Luo <yuan.luo@hotmail.com>
Date: Sat, 10 Jan 2026 07:03:47 +0800
Subject: [PATCH] [Fix CI] Fix test_mamba_unittest.py (#16810)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
---
 test/registered/attention/test_mamba_unittest.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/registered/attention/test_mamba_unittest.py b/test/registered/attention/test_mamba_unittest.py
index 0a0f45fb8..937c16baa 100644
--- 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):