self-supervised-learning/pyproject.toml at develop · UGA-BSAIL/self-supervised-learning · GitHub

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
[tool.poetry]
name = "self_supervised"
version = "0.1.0"
description = "Multi-object tracking for cotton flowers."
authors = ["Daniel Petti <djpetti@gmail.com>"]
license = "MIT"

[tool.poetry.dependencies]
python = "~3.10"
kedro = {extras = ["matplotlib", "pandas"], version = "^0.18.1"}
pandas = "1.5"
loguru = "^0.5.3"
seaborn = "^0.11.1"
matplotlib = "^3.3.4"
pydantic = "^1.8.1"
scipy = "^1.6.2"
numpy = "1.22"
opencv-contrib-python = "^4.5"
tqdm = "^4.64.0"
python-json-logger = "^2.0.4"
setuptools = "^65.7.0"
torch = "^2.2.1"
torchvision = "^0.17.1"
methodtools = "^0.4.5"
wandb = "^0.13.9"
torchmetrics = "^0.11.0"
scikit-learn = "^1.2.1"
segment-anything = {git = "https://github.com/facebookresearch/segment-anything.git"}
pyyaml = "^6"
ultralytics = { path = "external/ultralytics", develop = false }

[tool.poetry.dev-dependencies]
pytest = "^6.2.2"
pytest-mock = "^3.5.1"
pytest-black = "^0.3.12"
black = "^22.3.0"
isort = "^5.7.0"
pre-commit = "^2.11.0"
pytest-snapshot = "^0.5.0"
Faker = "^6.6.2"
pytest-coverage = "^0.0"
pytest-flake8 = "^1.0.7"
pydot = "^1.4.2"
jupyterlab = "^3.4.2"

[tool.black]
line-length = 79
exclude = '''
(
  /(
      \.eggs         # exclude a few common directories in the
    | \.git          # root of the project
    | \.venv
    | build
    | dist
  )/
)
'''

[tool.isort]
profile = "black"
line_length = 79
skip = "('.eggs', '.git', '.venv', 'build', 'dist')"

[tool.kedro]
package_name = "self_supervised"
project_name = "Self-Supervised Learning"
project_version = "0.18.1"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"