Skip to content

Commit 5b31610

Browse files
committed
update pyproject
1 parent 268c529 commit 5b31610

File tree

1 file changed

+141
-0
lines changed

1 file changed

+141
-0
lines changed

pyproject.toml

Lines changed: 141 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,141 @@
1+
[build-system]
2+
requires = ["maturin >= 1.8.6", "numpy >= 1.26.4"]
3+
build-backend = "maturin"
4+
5+
[project]
6+
name = "pylate-rs-cuda"
7+
version = "1.0.2"
8+
description = "Rust and WebAssembly library for late interaction models."
9+
authors = [
10+
{ name = "Raphael Sourty, LightOn", email = "raphael.sourty@lighton.ai" },
11+
]
12+
keywords = []
13+
classifiers = [
14+
"Programming Language :: Python :: 3",
15+
"Programming Language :: Rust",
16+
"Operating System :: OS Independent",
17+
]
18+
requires-python = ">=3.9"
19+
20+
dependencies = ["numpy >= 1.26.4"]
21+
22+
23+
[project.optional-dependencies]
24+
dev = [
25+
"maturin >= 1.8.6",
26+
"pytest-cov >= 5.0.0",
27+
"pytest >= 7.4.4",
28+
"ruff >= 0.1.15",
29+
"pre-commit >= 3.0.0",
30+
"pylate >= 1.2.0",
31+
"beir>=2.1.0",
32+
"ranx>=0.3.20",
33+
]
34+
35+
metal = []
36+
accelerate = []
37+
cuda = []
38+
cudnn = []
39+
mkl = []
40+
41+
42+
[project.urls]
43+
Homepage = "https://github.com/lightonai/pylate-rs"
44+
Documentation = "https://github.com/lightonai/pylate-rs"
45+
Repository = "https://github.com/lightonai/pylate-rs"
46+
47+
[tool.include]
48+
include = ["Cargo.toml", "pyproject.toml", "README.md", "src/*"]
49+
50+
[tool.maturin]
51+
bindings = "pyo3"
52+
python-source = "python"
53+
module-name = "pylate_rs.pylate_rs"
54+
features = ["tokenizers/onig", "hf-hub", "python", "cuda"]
55+
56+
[tool.pytest.ini_options]
57+
filterwarnings = [
58+
"ignore::DeprecationWarning",
59+
"ignore::RuntimeWarning",
60+
"ignore::UserWarning",
61+
]
62+
addopts = [
63+
"--doctest-modules",
64+
"--verbose",
65+
"-ra",
66+
"--cov-config=.coveragerc",
67+
"-m not web and not slow",
68+
]
69+
doctest_optionflags = ["NORMALIZE_WHITESPACE", "NUMBER"]
70+
norecursedirs = ["build", "docs", "node_modules"]
71+
markers = [
72+
"web: tests that require using the Internet",
73+
"slow: tests that take a long time to run",
74+
]
75+
76+
[tool.ruff]
77+
line-length = 88
78+
indent-width = 4
79+
target-version = "py310"
80+
81+
[tool.ruff.format]
82+
quote-style = "double"
83+
indent-style = "space"
84+
skip-magic-trailing-comma = false
85+
line-ending = "auto"
86+
docstring-code-format = false
87+
docstring-code-line-length = "dynamic"
88+
89+
[tool.ruff.lint]
90+
ignore = [
91+
"ANN001",
92+
"TID252",
93+
"FBT002",
94+
"ANN201",
95+
"G004",
96+
"BLE001",
97+
"S112",
98+
"PTH123",
99+
"S101",
100+
"ANN204",
101+
"PLR2004",
102+
"D104",
103+
"D100",
104+
"PTH118",
105+
"PTH110",
106+
"PTH103",
107+
"S311",
108+
"T201",
109+
"FBT001",
110+
"D107",
111+
"PTH107",
112+
"PTH109",
113+
"S603",
114+
"S607",
115+
"PGH004",
116+
"PLC0206",
117+
"PLW1508",
118+
"INP001",
119+
"FIX002",
120+
"TD003",
121+
"PGH003",
122+
"ANN401",
123+
"PERF401",
124+
"D203",
125+
"D213",
126+
"COM812",
127+
"N812",
128+
"PTH207",
129+
"PTH120",
130+
"SIM105",
131+
"PERF203",
132+
"TRY203",
133+
"TRY201",
134+
"PTH112",
135+
"B905",
136+
"UP037",
137+
]
138+
select = ["ALL"]
139+
fixable = ["ALL"]
140+
unfixable = []
141+
dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"

0 commit comments

Comments
 (0)