pyproject.toml 1.4 KB
Newer Older
1
[build-system]
2
# Should be mirrored in requirements-build.txt
3
requires = [
bnellnm's avatar
bnellnm committed
4
    "cmake>=3.21",
5
6
    "ninja",
    "packaging",
7
    "setuptools >= 49.4.0",
8
    "torch == 2.3.0",
9
10
11
    "wheel",
]
build-backend = "setuptools.build_meta"
12

13
14
15
[tool.ruff]
# Allow lines to be as long as 80.
line-length = 80
16
17
18
19
exclude = [
    # External file, leaving license intact
    "examples/fp8/quantizer/quantize.py"
]
20

21
22
23
24
25
26
27
28
29
30
31
32
33
34
[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    # Pyflakes
    "F",
    # pyupgrade
    # "UP",
    # flake8-bugbear
    "B",
    # flake8-simplify
    "SIM",
    # isort
    # "I",
35
    "G",
36
37
38
39
40
41
]
ignore = [
    # star imports
    "F405", "F403",
    # lambda expression assignment
    "E731",
42
43
    # Loop control variable not used within loop body
    "B007",
44
]
45
46
47
48
49

[tool.mypy]
python_version = "3.8"

ignore_missing_imports = true
50
51
check_untyped_defs = true
follow_imports = "skip"
52
53
54

files = "vllm"
# TODO(woosuk): Include the code from Megatron and HuggingFace.
55
56
exclude = [
    "vllm/model_executor/parallel_utils/|vllm/model_executor/models/",
57
58
    # Ignore triton kernels in ops.
    'vllm/attention/ops/.*\.py$'
59
]
60
61

[tool.codespell]
62
ignore-words-list = "dout, te, indicies, subtile"
63
skip = "./tests/prompts,./benchmarks/sonnet.txt,./tests/lora/data,./build"
64
65
66
67

[tool.isort]
use_parentheses = true
skip_gitignore = true
68
69
70
71

[tool.pytest.ini_options]
markers = [
    "skip_global_cleanup",
72
    "vlm: run tests for vision language models only",
73
]