Unverified Commit 830ed660 authored by Kris Hung's avatar Kris Hung Committed by GitHub
Browse files

fix: Skip kvbm consolidator test if vllm is not installed (#4007)

parent e5ae505b
...@@ -13,6 +13,7 @@ This test validates that: ...@@ -13,6 +13,7 @@ This test validates that:
""" """
import concurrent.futures import concurrent.futures
import importlib.util
import logging import logging
import os import os
import re import re
...@@ -25,12 +26,16 @@ import requests ...@@ -25,12 +26,16 @@ import requests
from tests.kvbm.common import ApiTester, check_logs_for_patterns from tests.kvbm.common import ApiTester, check_logs_for_patterns
from tests.utils.managed_process import ManagedProcess from tests.utils.managed_process import ManagedProcess
# Check if vLLM is available
HAS_VLLM = importlib.util.find_spec("vllm") is not None
# Test markers # Test markers
pytestmark = [ pytestmark = [
pytest.mark.kvbm, pytest.mark.kvbm,
pytest.mark.e2e, pytest.mark.e2e,
pytest.mark.slow, pytest.mark.slow,
pytest.mark.gpu_1, pytest.mark.gpu_1,
pytest.mark.skipif(not HAS_VLLM, reason="requires vllm"),
] ]
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment