Skip to content

Commit 830ed66

Browse files
authored
fix: Skip kvbm consolidator test if vllm is not installed (#4007)
1 parent e5ae505 commit 830ed66

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/kvbm/test_consolidator_router_e2e.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
"""
1414

1515
import concurrent.futures
16+
import importlib.util
1617
import logging
1718
import os
1819
import re
@@ -25,12 +26,16 @@
2526
from tests.kvbm.common import ApiTester, check_logs_for_patterns
2627
from tests.utils.managed_process import ManagedProcess
2728

29+
# Check if vLLM is available
30+
HAS_VLLM = importlib.util.find_spec("vllm") is not None
31+
2832
# Test markers
2933
pytestmark = [
3034
pytest.mark.kvbm,
3135
pytest.mark.e2e,
3236
pytest.mark.slow,
3337
pytest.mark.gpu_1,
38+
pytest.mark.skipif(not HAS_VLLM, reason="requires vllm"),
3439
]
3540

3641
logger = logging.getLogger(__name__)

0 commit comments

Comments
 (0)