diff --git a/tests/unit/models/generation/test_vllm_generation.py b/tests/unit/models/generation/test_vllm_generation.py index ba1bade3fc..aadb1fec77 100644 --- a/tests/unit/models/generation/test_vllm_generation.py +++ b/tests/unit/models/generation/test_vllm_generation.py @@ -43,15 +43,6 @@ } -@pytest.fixture(scope="module") -def check_vllm_available(): - """Skip tests if vLLM is not installed.""" - try: - import vllm # noqa: F401 - except ImportError: - pytest.skip("vLLM not installed") - - @pytest.fixture(scope="module") def cluster(): """Create a virtual cluster for testing.""" @@ -76,7 +67,7 @@ def tokenizer(): @pytest.fixture(scope="function") -def policy(cluster, tokenizer, check_vllm_available): +def policy(cluster, tokenizer): """Initialize the vLLM policy.""" # Create separate configs for each policy vllm_config = basic_vllm_test_config.copy() @@ -126,7 +117,7 @@ def test_input_data(tokenizer): ) -def test_vllm_missing_required_config_key(cluster, check_vllm_available): +def test_vllm_missing_required_config_key(cluster): """Test that an assertion error is raised when a required config key is missing.""" # Create a config missing a required key by removing 'model_name' incomplete_config = basic_vllm_test_config.copy()