-
Notifications
You must be signed in to change notification settings - Fork 46
/
Copy pathpyproject.toml
86 lines (72 loc) · 2.76 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
[tool.poetry]
name = "batchflow"
version = "0.8.10"
description = "ML pipelines, model configuration and batch management"
authors = ["Roman Kh <[email protected]>"]
license = "Apache License 2.0"
readme = "README.md"
documentation = "https://analysiscenter.github.io/batchflow/"
classifiers = [
'Development Status :: 4 - Beta',
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: Apache Software License',
'Operating System :: OS Independent',
'Programming Language :: Python',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.8',
'Topic :: Scientific/Engineering'
]
packages = [
{ include = "batchflow" },
]
[tool.poetry.dependencies]
python = "^3.8"
dill = "^0.3"
numpy = "^1.18"
numba = ">=0.56"
llvmlite = "*"
scipy = "^1.9"
tqdm = "^4.19"
# [tool.poetry.group.image.dependencies]
pillow = { version = ">=9.4,<11.0", optional = true }
matplotlib = { version = "^3.7", optional = true }
# [tool.poetry.group.research.dependencies]
multiprocess = { version = "^0.70", optional = true }
py-nbtools = { version = "*", optional = true }
# [tool.poetry.group.monitor.dependencies]
psutil = { version = "*", optional = true }
nvidia_smi = { version = "*", optional = true }
# [tool.poetry.group.nn.dependencies]
torch = { version = ">=1.13", optional = true }
torchvision = { version = ">=0.14", optional = true }
einops = { version = "^0.3", optional = true }
ptflops = { version = "^0.6", optional = true }
scikit-learn = { version = ">=0.21.3", optional = true }
# [tool.poetry.group.dev.dependencies]
pandas = { version = "^0.24", optional = true }
pytest = { version = "^7.0", optional = true }
pylint = { version = "^2.16", optional = true }
# [tool.poetry.group.jupyter.dependencies]
nbformat = { version = "*", optional = true }
nbconvert = { version = "*", optional = true }
ipykernel = { version = "*", optional = true }
ipython = { version = "*", optional = true }
notebook = { version = "*", optional = true }
jupyter_client = { version = "*", optional = true }
jupyter_server = { version = "*", optional = true }
requests = { version = "*", optional = true }
# [tool.poetry.group.other.dependencies]
urllib3 = { version = "^1.25", optional = true }
[tool.poetry.extras]
image = ["pillow", "matplotlib"]
nn = ["torch", "torchvision", "einops", "ptflops", "psutil"]
datasets = ["requests"]
profile = ["pandas", "psutil", "nvidia_smi", "matplotlib"]
jupyter = ["requests", "ipykernel", "ipython", "notebook", "nbconvert", "nbformat", "jupyter_client", "jupyter_server", "matplotlib"]
dev = ["pandas", "pytest", "pylint"]
research = ["pandas", "multiprocess", "psutil", "matplotlib"]
telegram = ["urllib3"]
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"