From 9c546bfd6df0d38d04894090277a12031fbaaaba Mon Sep 17 00:00:00 2001 From: Yikai Zhang Date: Wed, 12 Nov 2025 04:54:56 -0600 Subject: [PATCH] Fix the Wrong Return Type of `Scheduler.recv_requests` (#7886) --- python/sglang/srt/managers/scheduler.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 143d6cd52..263b2f7a5 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -24,7 +24,7 @@ from collections import deque from concurrent import futures from dataclasses import dataclass from http import HTTPStatus -from typing import Deque, Dict, List, Optional, Tuple, Union +from typing import Any, Deque, Dict, List, Optional, Tuple, Union import psutil import setproctitle @@ -1032,7 +1032,9 @@ class Scheduler( if envs.SGLANG_ENABLE_RUNTIME_MEM_LEAK_CHECK.get(): self._check_runtime_mem_leak() - def recv_requests(self) -> List[Req]: + def recv_requests( + self, + ) -> List[Union[TokenizedGenerateReqInput, TokenizedEmbeddingReqInput, Any]]: """Receive results at tp_rank = 0 and broadcast it to all other TP ranks.""" if self.recv_skipper is not None: