Localize cur_batch field in Scheduler to avoid field-based state access (#29407)

2026-07-10 08:55:06 +08:00
parent 69368d7593
commit 5be9c9f7c6
10 changed files with 60 additions and 48 deletions
@@ -193,7 +193,7 @@ class TestDecodeQueueCleanup(CustomTestCase):
        scheduler.dllm_manager = MagicMock()
        scheduler.dllm_manager.any_staging_reqs.return_value = False
        scheduler.last_batch = None
-        scheduler.cur_batch = None
+        scheduler.cur_batch_for_debug = None
        scheduler.enable_overlap = False
        scheduler.ps = SimpleNamespace(pp_size=1)
        scheduler.running_mbs = []