test_registry.py 273 Bytes
Newer Older
1
2
3
4
5
6
7
8
import pytest

from vllm.model_executor.models import _MODELS, ModelRegistry


@pytest.mark.parametrize("model_cls", _MODELS)
def test_registry_imports(model_cls):
    # Ensure all model classes can be imported successfully
9
    ModelRegistry.resolve_model_cls([model_cls])