-
Notifications
You must be signed in to change notification settings - Fork 3
/
pyproject.toml
99 lines (91 loc) · 2 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
[tool.poetry]
name = "alexandra_ai_eval"
version = "0.1.0"
description = "Evaluation of finetuned models."
authors = [
"Dan Saattrup Nielsen <[email protected]>",
"Anders Jess Pedersen <[email protected]>",
]
readme = "README.md"
license = "MIT"
[tool.poetry.dependencies]
python = ">=3.11,<3.13"
spacy = "^3.4.1"
sentencepiece = ">=0.1.96,<1.0.0"
protobuf = ">=3.0.0,<3.21.0"
tqdm = "^4.64.0"
seqeval = "^1.2.2"
huggingface-hub = ">=0.8.1,<1.0.0"
datasets = "^2.12.0"
codecarbon = "^2.1.3"
psutil = ">=5.9.0,<5.9.2"
fsspec = "^2023.9.2"
termcolor = "^2.3.0"
jiwer = "^3.0.3"
transformers = "^4.34.0"
evaluate = ">=0.4.0,<1.0.0"
pyctcdecode = ">=0.5.0,<1.0.0"
torch = "2.0.0"
tabulate = ">=0.9.0,<1.0.0"
requests = "^2.31.0"
kenlm = {url = "https://github.com/kpu/kenlm/archive/master.zip"}
gradio = "^3.41.0"
fastapi = ">=0.103.2,<1.0.0"
colorama = ">=0.4.6,<1.0.0"
llvmlite = ">=0.40.0,<1.0.0"
librosa = "^0.10.1"
urllib3 = "^2.0.7"
[tool.poetry.group.dev.dependencies]
pytest-xdist = "^3.3.1"
pytest-cov = "^4.1.0"
pre-commit = "^3.4.0"
readme-coverage-badger = ">=0.1.2,<1.0.0"
pytest = "^7.4.2"
pdoc = "^14.1.0"
[tool.poetry.scripts]
evaluate = "alexandra_ai_eval.cli:evaluate"
evaluate-gui = "alexandra_ai_eval.gui:main"
[tool.pytest.ini_options]
minversion = "7.0"
addopts = [
'--verbose',
'--durations=10',
'--color=yes',
'-s',
'-vv',
'--doctest-modules',
'--cov=src/alexandra_ai_eval',
'-n 8',
]
xfail_strict = true
filterwarnings = [
"error",
"ignore::UserWarning",
"ignore::DeprecationWarning",
"ignore::ImportWarning",
"ignore::FutureWarning",
]
log_cli_level = "info"
testpaths = ["tests", "src/alexandra_ai_eval"]
[tool.black]
line-length = 88
include = '\.pyi?$'
exclude = '''
/(
\.git
| \.hg
| \.mypy_cache
| \.tox
| \.venv
| _build
| buck-out
| build
)/
'''
[tool.ruff]
target-version = "py311"
[tool.ruff.extend-per-file-ignores]
"__init__.py" = ["F401"]
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"