-
Notifications
You must be signed in to change notification settings - Fork 86
/
Copy pathpyproject.toml
73 lines (63 loc) · 1.92 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
[tool.poetry]
name = "baal"
version = "2.0.0"
description = "Library to enable Bayesian active learning in your research or labeling work."
authors = ["Parmida Atighehchian <[email protected]>",
"Frederic Branchaud-Charron <[email protected]>",
"Jan Freyberg <[email protected]>",
"Lorne Schell <[email protected]>",
"Rafael Pardina <[email protected]>"
]
license = "Apache-2.0"
readme = "README.md"
documentation = "https://baal.readthedocs.io"
repository = "https://github.com/ElementAI/baal/"
[tool.poetry.dependencies]
python = ">=3.8,<4"
torch = ">=1.6.0"
torchmetrics = "^0.9.3"
h5py = "^3.4.0"
numpy = "^1.21.2"
matplotlib = "^3.4.3"
Pillow = ">=6.2.0"
tqdm = "^4.62.2"
structlog = "^22.0.0"
scikit-learn = "^1.0.0"
scipy = "^1.7.1"
# Vision
torchvision = { version = ">=0.7.0", optional=true }
lightning-flash = { version = ">=0.7.5", optional=true }
# NLP
transformers = {version = ">=4.10.2", optional=true}
accelerate = {version = "^0.28.0", optional=true}
datasets = {version = ">=2.14.6", optional=true}
jupyter = "^1.0.0"
[tool.poetry.dev-dependencies]
pytest = "^6.2.5"
pytest-cov = "^2.12.1"
torch-hypothesis = "0.2.0"
hypothesis = "4.24.0"
flake8 = "^3.9.2"
pytest-mock = "^3.6.1"
black = "^22.3.0"
# Issue with mypy https://github.com/pydantic/pydantic/issues/5192
mypy = "<=1.0.1"
bandit = "^1.7.1"
# Documentation
docutils = "0.16"
# Documentation
mkdocs-jupyter = "^0.24.0"
mkdocs-material = "^9.5.0"
Pygments = "^2.12.0"
mkdocstrings = {extras = ["python"], version = "^0.18.1"}
mkdocs-exclude-search = "^0.6.4"
[tool.poetry.extras]
vision = ["torchvision", "lightning-flash"]
nlp = ["transformers", "datasets", "accelerate"]
[tool.poetry.group.dev.dependencies]
neoteroi-mkdocs = "^1.0.5"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.black]
line-length = 100