-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
41 lines (36 loc) · 908 Bytes
/
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
[tool.poetry]
name = "ptb"
version = "0.1.0"
description = "Probabilistically True and Tight Bounds for Robust Deep Neural Network Training"
authors = ["ModarTensai <ModarTensai@gmail.com>"]
license = "MIT"
packages = [
{ include = "ptb", from = "." }
]
include = ["READEME.md"]
[tool.poetry.scripts]
ptb = 'ptb:cli.main'
[tool.poetry.dependencies]
python = "^3.7"
click = "^7.0"
torch = "^1.1"
torchvision = "^0.2.2"
foolbox = "^1.8.0"
tqdm = "^4.31.1"
matplotlib = "^3.0.3"
tb-nightly = {version = "^1.14.0a20190515", allows-prereleases = true}
ipykernel = { version = "^5.1.0", optional = true }
[tool.poetry.extras]
jupyter = ["ipykernel"]
[tool.poetry.dev-dependencies]
yapf = "^0.25.0"
pep8 = "^1.7.1"
pylint = "^2.3.1"
pydocstyle = "^3.0.0"
pytest = "^3.0"
pytest-cov = "^2.6.1"
tox = "^3.7.0"
mkdocs = "^1.0.4"
[build-system]
requires = ["poetry>=0.12"]
build-backend = "poetry.masonry.api"