-
Notifications
You must be signed in to change notification settings - Fork 5
/
pyproject.toml
126 lines (107 loc) · 3 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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
[tool.poetry]
name = "azimuth"
version = "2.8.0"
description = "Azimuth provides a unified error analysis experience to data scientists."
readme = "README.md"
authors = ["Azimuth team <[email protected]>"]
homepage = "https://servicenow.github.io/azimuth"
repository = "https://github.com/ServiceNow/azimuth"
packages = [{ include = "azimuth/" },]
[tool.poetry.dependencies]
python = ">=3.9,<3.10"
# Scientific
pandas = ">=1.0.5,<2,!=1.1.0"
baal = "1.5.2"
scikit-learn = "^1.0.1"
faiss-cpu = "=1.6.5"
tensorflow = "2.11.0"
tensorflow-text = "2.11.0"
tensorflow-estimator = "2.11.0"
onnx = "^1.12.0"
onnxruntime-extensions = "0.3.1"
onnxruntime = { version = "1.12.1", optional = true }
onnxruntime-gpu = { version = "1.12.1", optional = true }
scipy = "^1.10.0"
spectral-metric = "0.6.1"
torch = { version = "1.13.1", source = "PyPI" }
numpy = "1.23.5"
# NLP libraries
transformers = "4.16.2"
sentence-transformers = "2.1.0"
spacy = "3.1.5"
en_core_web_sm = {url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.1.0/en_core_web_sm-3.1.0.tar.gz#egg=en_core_web_sm"}
fr_core_news_md = {url = "https://github.com/explosion/spacy-models/releases/download/fr_core_news_md-3.1.0/fr_core_news_md-3.1.0.tar.gz#egg=fr_core_news_md"}
nlpaug = "1.1.10"
datasets = "2.1.0"
# Misc
filelock = "^3.0.12"
structlog = "21.1" # locked because 21.2 generates an error.
tqdm = "4.63.0"
plotly = "^5.3.1"
retrying = "^1.3.3"
jsonlines = "^3.1.0"
# FastAPI requirements
fastapi = "0.65.3"
orjson = "3.6.3"
pydantic = "^1.3"
uvicorn = "^0.13.4"
aiofiles = "^0.6.0"
# Dask dependencies
dask = "2021.12.0"
distributed = "2021.12.0"
[tool.poetry.extras]
cpu = ["onnxruntime"]
gpu = ["onnxruntime-gpu"]
[tool.poetry.group.dev.dependencies]
jupyterlab = "^3.0.5"
ipywidgets = ">=7.6"
matplotlib = "^3.4.1"
bokeh = "<3"
pyinstrument = "^4.3.0"
memory-profiler = "^0.61.0"
# Rare dependencies for specific experiments in notebooks
[tool.poetry.group.experimental]
optional = true
[tool.poetry.group.experimental.dependencies]
umap-learn = "^0.5.1"
hdbscan = "^0.8.27"
seaborn = "0.11.2"
[tool.poetry.group.test.dependencies]
pytest = "^6.0"
pytest-cov = "2.11.1"
pytest-env = "^0.6.2"
flake8 = ">=3.8"
flake8-black = "0.2.4"
black = ">=20.8b1"
mypy = "^0.910"
bandit = "^1.7.0"
pre-commit = "^2.15.0"
isort = "^5.9.3"
types-simplejson = "^3.17.0"
types-filelock = "^0.1.5"
types-orjson = "^3.6.0"
[tool.poetry.group.docs.dependencies]
mkdocs = "^1.2.3"
mkdocs-material = "^8.1.7"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.pytest.ini_options]
env = ["DASK_DISTRIBUTED__WORKER__DAEMON=False"]
filterwarnings = ["ignore::DeprecationWarning"]
[tool.black]
line-length = 100
[tool.isort]
profile = "black"
skip = ["__init__.py"]
filter_files = true
line_length = 100
[tool.mypy]
plugins = ["pydantic.mypy"]
python_version = 3.9
show_error_codes = true
warn_return_any = true
warn_unused_configs = true
warn_redundant_casts = true
warn_unused_ignores = true
ignore_missing_imports = true