diff --git a/test/registered/unit/server_args/test_server_args.py b/test/registered/unit/server_args/test_server_args.py index 79b85d49d..33d05e555 100644 --- a/test/registered/unit/server_args/test_server_args.py +++ b/test/registered/unit/server_args/test_server_args.py @@ -223,6 +223,7 @@ def test_megamoe_a2a_backend_parser_sets_ep_to_tp(): ] ) args = ServerArgs.from_cli_args(raw_args) + args._handle_a2a_moe() assert args.moe_a2a_backend == "megamoe" assert args.tp_size == 4 assert args.ep_size == 4