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/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -57,6 +57,7 @@ from sglang.srt.managers.io_struct import (
    AbortReq,
    CloseSessionReqInput,
    ExpertDistributionReq,
+    ExpertDistributionReqOutput,
    FlushCacheReq,
    GetInternalStateReq,
    GetInternalStateReqOutput,
@@ -1905,6 +1906,7 @@ class Scheduler(
            expert_distribution_recorder.dump_record()
        else:
            raise ValueError("Unrecognized ExpertDistributionReq value")
+        return ExpertDistributionReqOutput()

    def open_session(self, recv_req: OpenSessionReqInput):
        # handle error