diff --git a/tests/conftest.py b/tests/conftest.py index bbc56afdb..ea99b69ff 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -5,10 +5,13 @@ def pytest_collection_modifyitems(config, items): if sys.platform != "linux": + if not config.option.keyword or ( + config.option.keyword and "test_integration_vllm" in config.option.keyword + ): + print( + "WARNING: test_integration_vllm tests are skipped because vLLM only supports Linux platform (including WSL)." + ) skip_vllm = pytest.mark.skip(reason="vLLM models can only be run on Linux.") for item in items: if "test_integration_vllm" in item.nodeid: item.add_marker(skip_vllm) - print( - f"WARNING: {item.nodeid} is skipped because vLLM only supports Linux platform (including WSL)." - )