forked from illuin-tech/colpali
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
105 lines (91 loc) · 2.37 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
[project]
name = 'colpali_engine'
# dynamic = ["version"]
version = '0.0.1'
description = 'This repository centralizes ressources for the ColPali project.'
authors = [
{ name = 'Manuel Faysse', email = '[email protected]' },
{ name = 'Hugues Sibille', email = '[email protected]' },
{ name = 'Tony Wu', email = '[email protected]' },
]
readme = 'README.md'
requires-python = '>=3.9'
classifiers = [
'Intended Audience :: Science/Research',
'Intended Audience :: Developers',
'Operating System :: OS Independent',
'Private :: Do Not Upload',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.10',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Typing :: Typed',
]
dependencies = [
"torch>=2.2.0",
"transformers>=4.41.1",
"mteb>=1.12.22",
"requests",
"GPUtil",
"peft>=0.11.0, <0.12.0",
]
[project.optional-dependencies]
dev = [
"black>=24.4.2",
"coverage>=7.5.2",
"ipykernel>=6.29.4",
"mypy>=1.10.0",
"pytest>=8.2.1",
"ruff>=0.4.5",
]
train = [
"accelerate==0.30.1",
"configue==5.0.0",
"datasets==2.19.1",
"typer==0.12.3",
"bitsandbytes",
]
[project.urls]
homepage = "https://github.com/ManuelFay/colpali"
[build-system]
requires = ['setuptools', 'setuptools_scm[toml]', 'wheel']
build-backend = 'setuptools.build_meta'
[tool.setuptools_scm]
fallback_version = '0.0.0-dev'
[tool.setuptools]
zip-safe = false
platforms = ['any']
[tool.setuptools.packages.find]
include = ['colpali_engine', 'colpali_engine.*']
[tool.mypy]
check_untyped_defs = true
disallow_untyped_defs = true
enable_error_code = ['ignore-without-code']
exclude = ['docs/']
mypy_path = '$MYPY_CONFIG_FILE_DIR/typings'
no_implicit_optional = true
show_error_codes = true
warn_redundant_casts = true
warn_return_any = true
warn_unused_configs = true
warn_unused_ignores = true
warn_unreachable = true
[[tool.mypy.overrides]]
module = ['transformers', 'transformers.*', 'torch', 'torch.*']
ignore_missing_imports = true
[tool.coverage.run]
include = []
[tool.coverage.report]
exclude_lines = [
'pragma: no cover',
'raise NotImplementedError',
'if __name__ == "__main__":',
'if TYPE_CHECKING:',
'def __repr__',
]
[tool.black]
line-length = 120
[tool.ruff]
select = ["E", "F", "W", "I", "N"]
line-length = 120
[tool.ruff.per-file-ignores]
'__init__.py' = ["F401"]