[Bug] fix import and unit test (#25558)

Signed-off-by: Jonas M. Kübler <44084297+jmkuebler@users.noreply.github.com>
This commit is contained in:
Jonas M. Kübler 2025-09-24 12:17:59 +02:00 committed by GitHub
parent 42488dae69
commit 58c360d9be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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