-
Notifications
You must be signed in to change notification settings - Fork 21
/
pyproject.toml
112 lines (98 loc) · 2.81 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
[project]
name = "model2vec"
description = "The Fastest State-of-the-Art Static Embeddings in the World"
readme = { file = "README.md", content-type = "text/markdown" }
license = { file = "LICENSE" }
requires-python = ">=3.9"
authors = [{ name = "Stéphan Tulkens", email = "[email protected]"}, {name = "Thomas van Dongen", email = "[email protected]"}]
dynamic = ["version"]
classifiers = [
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Natural Language :: English",
]
dependencies = [
"jinja2",
"joblib",
"numpy",
"rich",
"safetensors",
"setuptools",
"tokenizers>=0.20",
"tqdm",
]
[build-system]
requires = ["setuptools>=64", "setuptools_scm>=8"]
build-backend = "setuptools.build_meta"
[tool.setuptools]
packages = ["model2vec"]
include-package-data = true
[tool.setuptools.package-data]
model2vec = ["assets/model_card_template.md"]
[project.optional-dependencies]
dev = [
"black",
"ipython",
"mypy",
"pre-commit",
"pytest",
"pytest-cov",
"ruff",
]
distill = ["torch", "transformers", "scikit-learn"]
onnx = ["onnx", "torch"]
[project.urls]
"Homepage" = "https://github.com/MinishLab"
"Bug Reports" = "https://github.com/MinishLab/model2vec/issues"
"Source" = "https://github.com/MinishLab/model2vec"
[tool.ruff]
exclude = [".venv/"]
line-length = 120
target-version = "py310"
[tool.ruff.lint]
select = [
# Annotations: Enforce type annotations
"ANN",
# Complexity: Enforce a maximum cyclomatic complexity
"C90",
# Pydocstyle: Enforce docstrings
"D",
# Isort: Enforce import order
"I",
# Numpy: Enforce numpy style
"NPY",
# Print: Forbid print statements
"T20",
]
ignore = [
# Allow self and cls to be untyped, and allow Any type
"ANN101", "ANN102", "ANN401",
# Pydocstyle ignores
"D100", "D101", "D104", "D203", "D212", "D401",
# Allow use of f-strings in logging
"G004"
]
[tool.pydoclint]
style = "sphinx"
exclude = "test_"
allow-init-docstring = true
arg-type-hints-in-docstring = false
check-return-types = false
require-return-section-when-returning-nothing = false
[tool.mypy]
python_version = "3.10"
warn_unused_configs = true
ignore_missing_imports = true
[tool.setuptools_scm]
# can be empty if no extra settings are needed, presence enables setuptools_scm
[tool.setuptools.dynamic]
version = {attr = "model2vec.version.__version__"}