mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 03:44:56 +08:00
10 lines
268 B
Python
10 lines
268 B
Python
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
|
|
ModelRegistry.load_model_cls(model_cls)
|