wehub-resource-sync
38 lines
952 B
Python
38 lines
952 B
Python
import sys
|
|
|
|
import pytest
|
|
|
|
# skip the test if vllm is installed
|
|
HAS_VLLM = False
|
|
try:
|
|
import vllm # noqa: F401
|
|
|
|
HAS_VLLM = True
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
@pytest.mark.skipif(not HAS_VLLM, reason="vllm is not installed")
|
|
def test_serve_llm_import_does_not_error():
|
|
# expected ImportError because of missing
|
|
# dependencies without ray[llm] dependencies
|
|
with pytest.raises(ImportError):
|
|
import ray.serve.llm # noqa: F401
|
|
with pytest.raises(ImportError):
|
|
from ray.serve.llm import (
|
|
LLMConfig, # noqa: F401
|
|
)
|
|
with pytest.raises(ImportError):
|
|
from ray.serve.llm.deployment import (
|
|
LLMServer, # noqa: F401
|
|
)
|
|
with pytest.raises(ImportError):
|
|
from ray.serve.llm import (
|
|
build_llm_deployment, # noqa: F401
|
|
build_openai_app, # noqa: F401
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-v", __file__]))
|