From cf08d8e659ae03e474373989bb13cba5501113d7 Mon Sep 17 00:00:00 2001 From: Sy03 <1370724210@qq.com> Date: Mon, 24 Aug 2026 05:21:10 +0800 Subject: [PATCH] [Bugfix][MiniCPM-o] Format streaming counter regression Signed-off-by: Sy03 <1370724210@qq.com> --- .../sched/test_omni_scheduler_streaming_input_counter.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/core/sched/test_omni_scheduler_streaming_input_counter.py b/tests/core/sched/test_omni_scheduler_streaming_input_counter.py index 3a7a29139..34529e836 100644 --- a/tests/core/sched/test_omni_scheduler_streaming_input_counter.py +++ b/tests/core/sched/test_omni_scheduler_streaming_input_counter.py @@ -62,9 +62,7 @@ def _make_scheduler(scheduler_cls, *, requests, running, waiting, counter, skipp def _upstream_num_unfinished(scheduler) -> int: """Mirror of ``Scheduler.get_num_unfinished_requests`` arithmetic.""" - num_waiting = ( - len(scheduler.waiting) + len(scheduler.skipped_waiting) - scheduler.num_waiting_for_streaming_input - ) + num_waiting = len(scheduler.waiting) + len(scheduler.skipped_waiting) - scheduler.num_waiting_for_streaming_input return num_waiting + len(scheduler.running) @@ -235,7 +233,9 @@ def test_ar_schedule_resyncs_before_delegating_upstream(monkeypatch: pytest.Monk seen: dict[str, int] = {} - monkeypatch.setattr(ar_sched_mod.OmniSchedulerMixin, "_consume_pending_connector_output", lambda self, model_mode: None) + monkeypatch.setattr( + ar_sched_mod.OmniSchedulerMixin, "_consume_pending_connector_output", lambda self, model_mode: None + ) monkeypatch.setattr(ar_sched_mod.OmniSchedulerMixin, "_process_pending_input_timeouts", lambda self: None) monkeypatch.setattr(OmniARScheduler, "_should_defer_waiting_admission", lambda self: False)