[mypy] Add mypy type annotation part 1 (#4006)

2024-04-13 06:35:50 +09:00
parent d4ec9ffb95
commit 09473ee41c
25 changed files with 171 additions and 72 deletions
--- a/vllm/transformers_utils/config.py
+++ b/vllm/transformers_utils/config.py
@@ -2,7 +2,8 @@ from typing import Dict, Optional

 from transformers import AutoConfig, PretrainedConfig

-from vllm.transformers_utils.configs import *
+from vllm.transformers_utils.configs import (ChatGLMConfig, DbrxConfig,
+                                             JAISConfig, MPTConfig, RWConfig)

 _CONFIG_REGISTRY: Dict[str, PretrainedConfig] = {
    "chatglm": ChatGLMConfig,