[mypy][5/N] Support all typing on model executor (#4427)

2024-04-29 11:01:26 +09:00
parent 03dd7d52bf
commit df29793dc7
10 changed files with 61 additions and 34 deletions
--- a/vllm/model_executor/guided_decoding/lm_format_enforcer_decoding.py
+++ b/vllm/model_executor/guided_decoding/lm_format_enforcer_decoding.py
@@ -61,6 +61,7 @@ def _normalize_json_schema_object(schema: Union[str, dict, BaseModel]) -> dict:
        return schema
    if isinstance(schema, BaseModel):
        return schema.model_json_schema()
+    raise AssertionError(f"Unsupported schema type {schema}")


@lru_cache