-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathpyproject.toml
More file actions
147 lines (142 loc) · 4.12 KB
/
pyproject.toml
File metadata and controls
147 lines (142 loc) · 4.12 KB
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
138
139
140
141
142
143
144
145
146
147
[project]
name = "validmind"
version = "2.13.0"
description = "ValidMind Library"
readme = "README.pypi.md"
requires-python = ">=3.9,<3.15"
license = { file = "LICENSE" }
authors = [
{ name = "Andres Rodriguez", email = "andres@validmind.ai" },
{ name = "Juan Martinez", email = "juan@validmind.ai" },
{ name = "Anil Sorathiya", email = "anil@validmind.ai" },
{ name = "Luis Pallares", email = "luis@validmind.ai" },
{ name = "John Walz", email = "john@validmind.ai" },
]
dependencies = [
"aiohttp[speedups] (<3.13.1)",
"ipywidgets",
"kaleido (>=0.2.1,!=0.2.1.post1,<1.0.0)",
"matplotlib",
"mistune (>=3.0.2,<4.0.0)",
"nest-asyncio (>=1.6.0,<2.0.0)",
"numpy (>=1.23,<2.0.0) ; python_version < '3.14'",
"numpy (>=2.3,<3.0.0) ; python_version >= '3.14'",
"openai (>=1)",
"pandas (>=2.0.3,<3.0.0)",
"plotly (>=6.0.0)",
"polars",
"python-dotenv",
"scikit-learn",
"seaborn",
"tabulate (>=0.9.0,<0.10.0)",
"tiktoken",
"tqdm",
"anywidget",
"beautifulsoup4",
]
[project.optional-dependencies]
all = [
"torch (>=2.0.0)",
"xgboost (>=1.5.2,<3)",
"transformers (>=4.32.0,<5.0.0)",
"pycocoevalcap",
"ragas (>=0.2.3,<=0.2.7)",
"langchain (>=0.2.0,<0.4.0)",
"langchain-community (>=0.2.0,<0.4.0)",
"sentencepiece (>=0.2.0,<0.3.0)",
"langchain-openai (>=0.1.8)",
"scipy",
"statsmodels (>=0.14.2,<0.15.0)",
"langdetect",
"nltk (>=3.8.1,<4.0.0)",
"textblob (>=0.18.0.post0,<0.19.0)",
"datasets (>=2.10.0,<3.0.0) ; python_version < '3.14'",
"datasets (>=4.4,<5.0.0) ; python_version >= '3.14'",
"evaluate",
"rouge (>=1)",
"bert-score (>=0.3.13)",
"pyarrow (<16) ; python_version < '3.13'",
"pyarrow (>=20,<21) ; python_version == '3.13'",
"pyarrow (>=22,<23) ; python_version >= '3.14'",
"arch (>=7.0.0)",
"shap (>=0.46.0)",
"scorecardpy==0.1.9.6",
"setuptools (<82)",
]
huggingface = [
"transformers (>=4.32.0,<5.0.0)",
"sentencepiece (>=0.2.0,<0.3.0)",
]
llm = [
"torch (>=2.0.0)",
"transformers (>=4.32.0,<5.0.0)",
"pycocoevalcap",
"ragas (>=0.2.3,<=0.2.7)",
"langchain (>=0.2.0,<0.4.0)",
"langchain-community (>=0.2.0,<0.4.0)",
"datasets (>=2.10.0,<3.0.0) ; python_version < '3.14'",
"datasets (>=4.4,<5.0.0) ; python_version >= '3.14'",
"sentencepiece (>=0.2.0,<0.3.0)",
"langchain-openai (>=0.1.8)",
"pyarrow (<16) ; python_version < '3.13'",
"pyarrow (>=20,<21) ; python_version == '3.13'",
"pyarrow (>=22,<23) ; python_version >= '3.14'",
"deepeval (>=3.7.0,<=3.7.5)",
]
nlp = [
"langdetect",
"nltk (>=3.8.1,<4.0.0)",
"textblob (>=0.18.0.post0,<0.19.0)",
"datasets (>=2.10.0,<3.0.0) ; python_version < '3.14'",
"datasets (>=4.4,<5.0.0) ; python_version >= '3.14'",
"transformers (>=4.32.0,<5.0.0)",
"evaluate",
"rouge (>=1)",
"bert-score (>=0.3.13)",
"pyarrow (<16) ; python_version < '3.13'",
"pyarrow (>=20,<21) ; python_version == '3.13'",
"pyarrow (>=22,<23) ; python_version >= '3.14'",
]
pytorch = ["torch (>=2.0.0)"]
stats = ["scipy", "statsmodels (>=0.14.2,<0.15.0)", "arch (>=7.0.0)"]
xgboost = ["xgboost (>=1.5.2,<3)"]
explainability = ["shap (>=0.46.0)"]
credit_risk = ["scorecardpy==0.1.9.6", "setuptools (<82)"]
datasets = [
"datasets (>=2.10.0,<3.0.0) ; python_version < '3.14'",
"datasets (>=4.4,<5.0.0) ; python_version >= '3.14'",
"pyarrow (<16) ; python_version < '3.13'",
"pyarrow (>=20,<21) ; python_version == '3.13'",
"pyarrow (>=22,<23) ; python_version >= '3.14'",
]
pii-detection = [
"presidio-analyzer ; python_version < '3.14'",
"presidio-analyzer (<2.2.360) ; python_version >= '3.14'",
"presidio-structured",
]
[tool.poetry.group.dev.dependencies]
black = "^22.1.0"
click = "*"
cython = "^0.29.34"
docstring_parser = "*"
flake8 = "^7.0.0"
griffe = "*"
ipykernel = "^6.22.0"
isort = "^5.12.0"
jupyter = "^1.0.0"
mdformat = "*"
papermill = "^2.4.0"
pdoc = "^14.4.0"
pre-commit = "^3.3.3"
sphinx = "^6.1.3"
sphinx-markdown-builder = "^0.5.5"
sphinx-rtd-theme = "^1.2.0"
twine = "^4.0.2"
[project.scripts]
vm-create-new-test = "scripts.create_new_test:generate_test"
[build-system]
build-backend = "poetry.core.masonry.api"
requires = ["poetry-core>=1.9.0"]
[tool.isort]
known_first_party = "validmind"
profile = "black"