easydistill/pyproject.toml at main · modelscope/easydistill · GitHub

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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "easydistill"
version = "2.0.0"
description = "EasyDistill 2 - Upgraded knowledge distillation toolkit for LLMs."
readme = "README.md"
license = {text = "Apache-2.0"}
requires-python = ">=3.9"
authors = [
    {name = "PAI Team"},
]
keywords = ["distillation", "LLM", "SFT", "data-synthesis", "PAI"]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: Apache Software License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]

dependencies = [
    "pydantic>=2.0",
    "typing-extensions>=4.5.0",
    "httpx>=0.27.0",
    "tqdm>=4.66.0",
    "pyyaml>=6.0",
    "openai>=1.30.0",
    "python-dotenv>=1.0.0",
    "langgraph>=0.2.0",
    "langchain-core>=0.3.0",
]

[project.optional-dependencies]
local = ["vllm>=0.5.0", "transformers>=4.40.0", "jinja2>=3.1.0"]
t2i = ["dashscope>=1.20.0"]
t2v = ["opencv-python-headless>=4.9.0"]
all = [
    "vllm>=0.5.0",
    "transformers>=4.40.0",
    "jinja2>=3.1.0",
    "dashscope>=1.20.0",
    "opencv-python-headless>=4.9.0",
]
dev = [
    "pytest>=8.0",
    "pytest-asyncio>=0.23.0",
    "black>=24.0",
    "ruff>=0.4.0",
    "mypy>=1.9.0,<2",
    "types-PyYAML>=6.0",
]

[project.urls]
Homepage = "https://github.com/modelscope/easydistill"
Repository = "https://github.com/modelscope/easydistill"

[project.scripts]
easydistill = "easydistill.cli:main"

[tool.setuptools.packages.find]
where = ["."]
include = ["easydistill*"]

[tool.black]
line-length = 100
target-version = ['py39']

[tool.ruff]
line-length = 100

[tool.ruff.lint]
select = ["E", "F", "I", "W", "UP", "B", "SIM", "C4"]
ignore = [
    # Allow legacy typing imports for Python 3.9 compatibility.
    "UP006",
    "UP007",
    "UP035",
]

[tool.mypy]
python_version = "3.9"
warn_return_any = true
warn_unused_configs = true
ignore_missing_imports = true

[[tool.mypy.overrides]]
module = "click.*"
follow_imports = "skip"

# anyio (pulled in via httpx) ships py.typed sources using match statements,
# which mypy rejects under python_version = "3.9".
[[tool.mypy.overrides]]
module = "anyio.*"
follow_imports = "skip"
follow_imports_for_stubs = true

# pytest (pulled in via langsmith's optional test integration) also ships
# py.typed sources using match statements; skip it for the same reason.
[[tool.mypy.overrides]]
module = "pytest.*"
follow_imports = "skip"
follow_imports_for_stubs = true

# huggingface_hub (pulled in via transformers / diffusers / trl) ships
# py.typed sources using pattern matching, which mypy rejects under
# python_version = "3.9".
[[tool.mypy.overrides]]
module = "huggingface_hub.*"
follow_imports = "skip"
follow_imports_for_stubs = true

# torch ships py.typed sources using pattern matching, which mypy rejects
# under python_version = "3.9".
[[tool.mypy.overrides]]
module = "torch.*"
follow_imports = "skip"
follow_imports_for_stubs = true

# matplotlib ships py.typed sources using pattern matching, which mypy rejects
# under python_version = "3.9".
[[tool.mypy.overrides]]
module = "matplotlib.*"
follow_imports = "skip"
follow_imports_for_stubs = true

# numpy ships stubs using `type` statements, which mypy rejects under
# python_version = "3.9".
[[tool.mypy.overrides]]
module = "numpy.*"
follow_imports = "skip"
follow_imports_for_stubs = true