Tidy vllm/config/__init__.py to only add classes and functions (#26405)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-10-08 15:10:00 +01:00
committed by GitHub
parent 338b1bf04f
commit 2f99f2f506
18 changed files with 46 additions and 77 deletions

View File

@@ -1,42 +1,28 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from vllm.config.cache import (
BlockSize,
CacheConfig,
CacheDType,
MambaDType,
PrefixCachingHashAlgo,
)
from vllm.config.cache import CacheConfig
from vllm.config.compilation import (
CompilationConfig,
CompilationLevel,
CUDAGraphMode,
PassConfig,
)
from vllm.config.device import Device, DeviceConfig
from vllm.config.device import DeviceConfig
from vllm.config.kv_events import KVEventsConfig
from vllm.config.kv_transfer import KVTransferConfig
from vllm.config.load import LoadConfig
from vllm.config.lora import LoRAConfig
from vllm.config.model import (
ConvertOption,
HfOverrides,
LogprobsMode,
ModelConfig,
ModelDType,
ModelImpl,
RunnerOption,
TaskOption,
TokenizerMode,
iter_architecture_defaults,
try_match_architecture_defaults,
)
from vllm.config.multimodal import MMCacheType, MMEncoderTPMode, MultiModalConfig
from vllm.config.observability import DetailedTraceModules, ObservabilityConfig
from vllm.config.parallel import DistributedExecutorBackend, EPLBConfig, ParallelConfig
from vllm.config.multimodal import MultiModalConfig
from vllm.config.observability import ObservabilityConfig
from vllm.config.parallel import EPLBConfig, ParallelConfig
from vllm.config.pooler import PoolerConfig
from vllm.config.scheduler import RunnerType, SchedulerConfig, SchedulerPolicy
from vllm.config.scheduler import SchedulerConfig
from vllm.config.speculative import SpeculativeConfig
from vllm.config.speech_to_text import SpeechToTextConfig
from vllm.config.structured_outputs import StructuredOutputsConfig
@@ -56,20 +42,17 @@ from vllm.config.vllm import (
set_current_vllm_config,
)
# __all__ should only contain classes and functions.
# Types and globals should be imported from their respective modules.
__all__ = [
# From vllm.config.cache
"BlockSize",
"CacheConfig",
"CacheDType",
"MambaDType",
"PrefixCachingHashAlgo",
# From vllm.config.compilation
"CompilationConfig",
"CompilationLevel",
"CUDAGraphMode",
"PassConfig",
# From vllm.config.device
"Device",
"DeviceConfig",
# From vllm.config.kv_events
"KVEventsConfig",
@@ -80,34 +63,20 @@ __all__ = [
# From vllm.config.lora
"LoRAConfig",
# From vllm.config.model
"ConvertOption",
"HfOverrides",
"LogprobsMode",
"ModelConfig",
"ModelDType",
"ModelImpl",
"RunnerOption",
"TaskOption",
"TokenizerMode",
"iter_architecture_defaults",
"try_match_architecture_defaults",
# From vllm.config.multimodal
"MMCacheType",
"MMEncoderTPMode",
"MultiModalConfig",
# From vllm.config.observability
"DetailedTraceModules",
"ObservabilityConfig",
# From vllm.config.parallel
"DistributedExecutorBackend",
"EPLBConfig",
"ParallelConfig",
# From vllm.config.pooler
"PoolerConfig",
# From vllm.config.scheduler
"RunnerType",
"SchedulerConfig",
"SchedulerPolicy",
# From vllm.config.speculative
"SpeculativeConfig",
# From vllm.config.speech_to_text