Fix test_expert_distribution failure (#4752)

2025-03-25 16:17:03 +08:00
parent e45ae444db
commit eb934bdf3b
5 changed files with 41 additions and 17 deletions
--- a/python/sglang/srt/managers/io_struct.py
+++ b/python/sglang/srt/managers/io_struct.py
@@ -664,6 +664,11 @@ class ExpertDistributionReq(Enum):
    DUMP_RECORD = 3


+@dataclass
+class ExpertDistributionReqOutput:
+    pass
+
+
@dataclass
 class ProfileReq:
    type: ProfileReqType