Skip to content

Commit 643f502

Browse files
kevincheng2claude
andcommitted
fix(test): fix unit test errors for _trigger_preempt and wakeup with MTP
- Use BatchRequest instead of list in test_trigger_preempt_records_tasks - Add missing enable_cache_manager_v1 attr in TestSleepWakeupBehavior._make_runner Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent c5e3620 commit 643f502

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

tests/v1/test_resource_manager_v1.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from fastdeploy.config import CacheConfig, FDConfig, ParallelConfig, SchedulerConfig
3131
from fastdeploy.engine.args_utils import EngineArgs
3232
from fastdeploy.engine.request import (
33+
BatchRequest,
3334
CompletionOutput,
3435
ImagePosition,
3536
Request,
@@ -683,12 +684,12 @@ def test_trigger_preempt_records_tasks(self):
683684
manager.running = [request, preempted_req]
684685

685686
preempted_reqs = []
686-
scheduled_reqs = []
687-
can_schedule = manager._trigger_preempt(request, 2, preempted_reqs, scheduled_reqs)
687+
batch_request = BatchRequest()
688+
can_schedule = manager._trigger_preempt(request, 2, preempted_reqs, batch_request)
688689
self.assertTrue(can_schedule)
689690
self.assertIn(preempted_req.request_id, manager.to_be_rescheduled_request_id_set)
690691
self.assertEqual(preempted_reqs[0], preempted_req)
691-
self.assertEqual(scheduled_reqs[0].request_id, preempted_req.request_id)
692+
self.assertEqual(batch_request.requests[0].request_id, preempted_req.request_id)
692693

693694
def test_available_position_and_real_bsz(self):
694695
manager = _build_manager()

tests/worker/test_gpu_model_runner.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -484,6 +484,7 @@ def _make_runner(self):
484484
runner.is_kvcache_sleeping = False
485485
runner.use_cudagraph = False
486486
runner.spec_method = None
487+
runner.enable_cache_manager_v1 = False
487488
runner.local_rank = 0
488489
runner.device_id = 1
489490
runner.num_gpu_blocks = 8

0 commit comments

Comments
 (0)