-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
117 lines (105 loc) · 2.36 KB
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
[project]
name = "birr"
version = "0.2.0"
description = "Tool for orchestrating LLM/VLM inference"
authors = [{ name = "Allen AI", email = "[email protected]" }]
license = { text = "Apache-2.0" }
readme = "README.md"
requires-python = ">=3.8"
dependencies = [
"jsonschema",
"pillow",
"pyaml",
"pydantic>=2,<3",
"pydantic-settings",
"zstandard",
"click",
]
classifiers = ["Development Status :: 3 - Alpha", "Typing :: Typed"]
[project.urls]
Homepage = "https://github.com/allenai/birr"
Source = "https://github.com/allenai/birr"
Tracker = "https://github.com/allenai/birr/issues"
[project.optional-dependencies]
dev = [
"black[jupyter]>=22.6.0",
"flake8>=5.0",
"flake8-pyi>=22.8.1",
"Flake8-pyproject>=1.1.0",
"ipdb>=0.13.0",
"ipython>=8.4.0",
"isort>=5.10.1",
"mypy==1.13.0",
"pytest>=5.2",
"requests"
]
batch_inference = [
"bitsandbytes",
"numpy",
"nvidia-ml-py", # gets pynvml
"qwen-vl-utils",
"ray[default]",
"sentencepiece",
"torch>=2.2.0",
"transformers>=4.45.1",
]
# vllm is separated because it cannot be installed on macs
vllm = [
"vllm==0.6.3.post1",
"vllm-flash-attn==2.6.1",
]
[build-system]
build-backend = "setuptools.build_meta"
requires = ["setuptools >= 61.0.0", "wheel"]
[tool.setuptools.packages.find]
where = ["src"]
[tool.setuptools.package-data]
birr = ["py.typed", "*.pyi", "demo/res/*/*"]
[tool.black]
line-length = 115
include = '\.pyi?$'
exclude = '''
(
__pycache__
| \.git
| \.mypy_cache
| \.pytest_cache
| \.vscode
| \.venv
| \bdist\b
| \bdoc\b
| \.cargo
| configs
| docs
| scripts
| sources
| target
| tests/config
| tests/data
| tests/work
)
'''
target-version = ["py38", "py39", "py310", "py311", "py312"]
[tool.mypy]
python_version = "3.8"
ignore_missing_imports = true
no_site_packages = true
allow_redefinition = false
warn_unused_configs = true
warn_unused_ignores = true
warn_no_return = true
warn_return_any = false
warn_unreachable = true
show_error_codes = true
pretty = true
plugins = ["numpy.typing.mypy_plugin"]
[tool.mypy-tests]
strict_optional = false
[tool.pytest.ini_options]
testpaths = ['tests/']
pythonpath = ['.', 'src']
python_classes = ['Test*', '*Test']
log_format = '%(asctime)s - %(levelname)s - %(name)s - %(message)s'
log_level = 'DEBUG'
markers = []
filterwarnings = []