Skip to content

Commit f089961

Browse files
Isotr0pyFeiDaLI
authored andcommitted
[CI/Build] Fix disabled v1 attention backend selection test (vllm-project#25471)
Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn>
1 parent 351e121 commit f089961

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/kernels/attention/test_attention_selector.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,6 @@ def generate_params():
6767
return params
6868

6969

70-
@pytest.mark.skip(reason="Skipped for now. Should be revisited.")
7170
@pytest.mark.parametrize("device, name, use_mla, block_size",
7271
generate_params())
7372
def test_env(
@@ -189,7 +188,7 @@ def test_env(
189188
# FlashMLA only supports block_size == 64
190189
pytest.skip("FlashMLA only supports block_size 64")
191190
else:
192-
from vllm.attention.backends.flashmla import (
191+
from vllm.v1.attention.backends.mla.flashmla import ( # noqa: E501
193192
is_flashmla_supported)
194193
is_supported, _ = is_flashmla_supported()
195194
if not is_supported:

0 commit comments

Comments
 (0)