pyproject.toml 5.16 KB
Newer Older
1
[build-system]
2
# Should be mirrored in requirements/build.txt
3
requires = [
4
    "cmake>=3.26",
5
    "ninja",
6
7
    "packaging>=24.2",
    "setuptools>=77.0.3,<80.0.0",
8
    "setuptools-scm>=8.0",
Huy Do's avatar
Huy Do committed
9
    "torch == 2.7.0",
10
    "wheel",
11
    "jinja2",
12
13
]
build-backend = "setuptools.build_meta"
14

15
16
17
[project]
name = "vllm"
authors = [{name = "vLLM Team"}]
18
19
license = "Apache-2.0"
license-files = ["LICENSE"]
20
21
22
23
24
25
26
27
28
29
30
31
32
readme = "README.md"
description = "A high-throughput and memory-efficient inference and serving engine for LLMs"
classifiers = [
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Intended Audience :: Developers",
    "Intended Audience :: Information Technology",
    "Intended Audience :: Science/Research",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Information Analysis",
]
Martin Hoyer's avatar
Martin Hoyer committed
33
requires-python = ">=3.9,<3.13"
34
35
36
37
38
39
40
41
42
43
dynamic = [ "version", "dependencies", "optional-dependencies"]

[project.urls]
Homepage="https://github.com/vllm-project/vllm"
Documentation="https://vllm.readthedocs.io/en/latest/"
Slack="http://slack.vllm.ai/"

[project.scripts]
vllm = "vllm.entrypoints.cli.main:main"

44
45
46
[project.entry-points."vllm.general_plugins"]
lora_filesystem_resolver = "vllm.plugins.lora_resolvers.filesystem_resolver:register_filesystem_resolver"

47
[tool.setuptools_scm]
48
# no extra settings needed, presence enables setuptools-scm
49
50
51

[tool.setuptools.packages.find]
where = ["."]
52
include = ["vllm*"]
53

54
55
56
57
58
[tool.yapfignore]
ignore_patterns = [
    "build/**",
]

59
60
61
[tool.ruff]
# Allow lines to be as long as 80.
line-length = 80
62
63
exclude = [
    # External file, leaving license intact
64
65
    "examples/other/fp8/quantizer/quantize.py",
    "vllm/vllm_flash_attn/flash_attn_interface.pyi"
66
]
67

68
[tool.ruff.lint.per-file-ignores]
69
"vllm/third_party/**" = ["ALL"]
70
71
"vllm/version.py" = ["F401"]
"vllm/_version.py" = ["ALL"]
72
73
74
75
76
77
78
79
80
81
# Python 3.8 typing. TODO: Remove these excludes after v1.0.0
"vllm/adapter_commons/**/*.py" = ["UP006", "UP035"]
"vllm/attention/**/*.py" = ["UP006", "UP035"]
"vllm/compilation/**/*.py" = ["UP006", "UP035"]
"vllm/core/**/*.py" = ["UP006", "UP035"]
"vllm/device_allocator/**/*.py" = ["UP006", "UP035"]
"vllm/distributed/**/*.py" = ["UP006", "UP035"]
"vllm/engine/**/*.py" = ["UP006", "UP035"]
"vllm/executor/**/*.py" = ["UP006", "UP035"]
"vllm/lora/**/*.py" = ["UP006", "UP035"]
82
83
84
"vllm/model_executor/layers/**/*.py" = ["UP006", "UP035"]
"vllm/model_executor/model_loader/**/*.py" = ["UP006", "UP035"]
"vllm/model_executor/models/**/*.py" = ["UP006", "UP035"]
85
86
87
88
89
90
91
92
93
"vllm/platforms/**/*.py" = ["UP006", "UP035"]
"vllm/plugins/**/*.py" = ["UP006", "UP035"]
"vllm/profiler/**/*.py" = ["UP006", "UP035"]
"vllm/prompt_adapter/**/*.py" = ["UP006", "UP035"]
"vllm/spec_decode/**/*.py" = ["UP006", "UP035"]
"vllm/transformers_utils/**/*.py" = ["UP006", "UP035"]
"vllm/triton_utils/**/*.py" = ["UP006", "UP035"]
"vllm/vllm_flash_attn/**/*.py" = ["UP006", "UP035"]
"vllm/worker/**/*.py" = ["UP006", "UP035"]
94
"vllm/utils.py" = ["UP006", "UP035"]
95

96
97
98
99
100
101
102
[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    # Pyflakes
    "F",
    # pyupgrade
103
    "UP",
104
105
106
107
108
109
    # flake8-bugbear
    "B",
    # flake8-simplify
    "SIM",
    # isort
    # "I",
110
    "G",
111
112
113
114
115
116
]
ignore = [
    # star imports
    "F405", "F403",
    # lambda expression assignment
    "E731",
117
118
    # Loop control variable not used within loop body
    "B007",
119
120
    # f-string format
    "UP032",
121
122
    # Can remove once 3.10+ is the minimum Python version
    "UP007",
123
]
124
125
126

[tool.mypy]
ignore_missing_imports = true
127
check_untyped_defs = true
128
follow_imports = "silent"
129

130
# After fixing type errors resulting from follow_imports: "skip" -> "silent",
131
# move the directory here and remove it from tools/mypy.sh
132
133
134
135
files = [
    "vllm/*.py",
    "vllm/adapter_commons",
    "vllm/assets",
136
    "vllm/entrypoints",
137
    "vllm/core",
138
    "vllm/inputs",
139
    "vllm/logging_utils",
140
141
142
143
144
145
    "vllm/multimodal",
    "vllm/platforms",
    "vllm/transformers_utils",
    "vllm/triton_utils",
    "vllm/usage",
]
146
# TODO(woosuk): Include the code from Megatron and HuggingFace.
147
148
exclude = [
    "vllm/model_executor/parallel_utils/|vllm/model_executor/models/",
149
150
    # Ignore triton kernels in ops.
    'vllm/attention/ops/.*\.py$'
151
]
152
153

[tool.codespell]
154
ignore-words-list = "dout, te, indicies, subtile, ElementE"
155
skip = "tests/models/fixtures/*,tests/prompts/*,benchmarks/sonnet.txt,tests/lora/data/*,build/*,vllm/third_party/*"
156
157
158
159

[tool.isort]
use_parentheses = true
skip_gitignore = true
160
161
162
163

[tool.pytest.ini_options]
markers = [
    "skip_global_cleanup",
164
165
    "core_model: enable this model test in each PR instead of only nightly",
    "cpu_model: enable this model test in CPU tests",
166
167
    "split: run this test as part of a split",
    "distributed: run this test only in distributed GPU tests",
Joe Runde's avatar
Joe Runde committed
168
    "skip_v1: do not run this test with v1",
169
    "optional: optional tests that are automatically skipped, include --optional to run them",
170
]
171
172

[tool.pymarkdown]
173
plugins.md004.style = "sublist" # ul-style
174
175
176
177
plugins.md013.enabled = false # line-length
plugins.md041.enabled = false # first-line-h1
plugins.md033.enabled = false # inline-html
plugins.md024.allow_different_nesting = true # no-duplicate-headers