diff --git a/python/sglang/srt/managers/schedule_batch.py b/python/sglang/srt/managers/schedule_batch.py index 0e8c362be..53f593988 100755 --- a/python/sglang/srt/managers/schedule_batch.py +++ b/python/sglang/srt/managers/schedule_batch.py @@ -1514,10 +1514,6 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin): # Forward-pass metrics fpm_start_time: float = 0.0 - # Whether to return captured experts - return_routed_experts: bool = False - return_indexer_topk: bool = False - # hicache pointer for synchronizing data loading from CPU to GPU hicache_consumer_index: int = -1 @@ -1668,8 +1664,6 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin): device=req_to_token_pool.device, spec_algorithm=spec_algorithm, return_hidden_states=any(req.return_hidden_states for req in reqs), - return_routed_experts=any(req.return_routed_experts for req in reqs), - return_indexer_topk=any(req.return_indexer_topk for req in reqs), is_prefill_only=all(req.is_prefill_only for req in reqs), chunked_req=chunked_req, dllm_config=dllm_config, diff --git a/test/registered/8-gpu-models/test_return_indexer_topk.py b/test/registered/rl/test_return_indexer_topk.py similarity index 100% rename from test/registered/8-gpu-models/test_return_indexer_topk.py rename to test/registered/rl/test_return_indexer_topk.py