From e1cdd0c23bb3cb589479a6ea47d1c920d8006059 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=AA=E5=BF=97=E9=B9=8F?= Date: Wed, 22 Apr 2026 12:36:20 +0800 Subject: [PATCH] [BugFix][CI]: Fix test_omni_sleep_mode ci bug (#3010) Signed-off-by: princepride --- tests/e2e/offline_inference/test_omni_sleep_mode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/offline_inference/test_omni_sleep_mode.py b/tests/e2e/offline_inference/test_omni_sleep_mode.py index 2ad0b53b0..fc4001ff5 100644 --- a/tests/e2e/offline_inference/test_omni_sleep_mode.py +++ b/tests/e2e/offline_inference/test_omni_sleep_mode.py @@ -123,7 +123,7 @@ async def test_multistage_sleep_h100(tp_size): pass acks = await engine.sleep(stage_ids=[0, 1], level=2) - assert len(acks) == 2 * tp_size + assert len(acks) == 2 await engine.wake_up(stage_ids=[0, 1]) async for _ in engine.generate("verify", sampling_params=[SamplingParams(), sp]):