From f98d4d38c00ac0f768e7f1a87e8f2807badbe5e2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jonas=20M=2E=20K=C3=BCbler?= <44084297+jmkuebler@users.noreply.github.com> Date: Wed, 24 Sep 2025 12:17:59 +0200 Subject: [PATCH] [Bug] fix import and unit test (#25558) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Jonas M. Kübler <44084297+jmkuebler@users.noreply.github.com> Signed-off-by: yewentao256 --- tests/v1/attention/test_attention_splitting.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/v1/attention/test_attention_splitting.py b/tests/v1/attention/test_attention_splitting.py index 2fd11415d4904..d81f3da7e9cd9 100644 --- a/tests/v1/attention/test_attention_splitting.py +++ b/tests/v1/attention/test_attention_splitting.py @@ -11,7 +11,7 @@ from vllm.v1.attention.backends.utils import (UBatchSlice, slice_query_start_locs, split_attn_metadata, split_decodes_and_prefills) -from vllm.v1.worker.ubatch_utils import create_ubatch_slices +from vllm.v1.worker.ubatch_splitting import create_ubatch_slices @pytest.fixture