-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
41 lines (34 loc) · 856 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 = "pytorch-ssd"
version = "1.8.0"
description = "Single Shot Multi-Box Detector implementation in PyTorch"
authors = ["PiotrJZielinski <piotrekzie100@gmail.com>"]
[tool.poetry.dependencies]
python = "^3.8"
torch = "^1.7.1"
torchvision = "^0.8.2"
pytorch-lightning = "^1.1.2"
albumentations = "^0.5.2"
numpy = "^1.19.3"
h5py = "^3.2.1"
pycocotools = "^2.0.2"
pillow = "^8.0.1"
wandb = "^0.10.12"
[tool.poetry.dev-dependencies]
pytest = "^6.1.2"
pytest-cov = "^2.10.1"
ipython = "^7.19.0"
[tool.poetry.scripts]
ssd = "pytorch_ssd.main:cli"
[build-system]
requires = ["poetry>=0.12"]
build-backend = "poetry.masonry.api"
[tool.flakehell]
max-line-length = 88
ignore = ["E203", "E231"]
[tool.pytest.ini_options]
addopts = "--color=yes --ignore=data"
[tool.isort]
line_length = 88
multi_line_output = 3
include_trailing_comma = "True"