-
Notifications
You must be signed in to change notification settings - Fork 5
/
pyproject.toml
97 lines (88 loc) · 2.52 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
87
88
89
90
91
92
93
94
95
96
97
[tool.poetry]
name = "ocl"
version = "0.1.0"
description = ""
authors = [
"Max Horn <[email protected]>",
"Maximilian Seitzer <[email protected]>",
"Andrii Zadaianchuk <[email protected]>",
"Zixu Zhao <[email protected]>",
"Dominik Zietlow <[email protected]>",
"Florian Wenzel <[email protected]>",
"Tianjun Xiao <[email protected]>"
]
license = "Apache-2.0"
repository = "https://github.com/amazon-science/object-centric-learning-framework"
readme = "README.md"
include = ["configs/"]
packages = [
{ include = "ocl" },
{ include = "routed" },
]
[tool.poetry.scripts]
ocl_train = "ocl.cli.train:train"
ocl_eval = "ocl.cli.eval:evaluate"
ocl_compute_dataset_size = "ocl.cli.compute_dataset_size:compute_size"
[tool.poetry.dependencies]
python = ">=3.8.1,<3.9"
# There seems to be an issue in torch 1.12.x with masking and multi-head
# attention. This prevents the usage of makes without a batch dimension.
# Staying with torch 1.11.x version for now.
torch = "^2.0.0"
pytorch-lightning = "^2.0.0"
hydra-zen = "^0.7.0"
torchtyping = "^0.1.4"
hydra-core = "^1.3.0"
torchvision = "^0.15.1"
torchmetrics = "^0.11.1"
matplotlib = "^3.7.1"
moviepy = "^1.0.3"
scipy = "<=1.8"
awscli = "^1.22.90"
scikit-learn = "^1.0.2"
pyamg = "^4.2.3"
botocore = { extras = ["crt"], version = "^1.27.22" }
motmetrics = "^1.2.5"
einops = "^0.6.0"
torchdata = "^0.6.0"
braceexpand = "^0.1.7"
s3fs = "^2022.11.0"
decord = "^0.6.0"
# Dependencies for support of externally defined models.
timm = {version = "0.6.13", optional = true}
clip = {git = "https://github.com/openai/CLIP.git", rev = "main", optional = true}
ftfy = {version = "^6.1.1", optional = true}
regex = {version = "^2022.7.9", optional = true}
[tool.poetry.group.dev.dependencies]
black = "^22.1.0"
pytest = "^7.0.1"
flake8 = "^6.0.0"
flake8-isort = "^6.0.0"
pre-commit = "^2.17.0"
flake8-tidy-imports = "^4.7"
flake8-bugbear = "^23.3"
flake8-docstrings = "^1.7.0"
[tool.poetry.group.docs]
optional = true
[tool.poetry.group.docs.dependencies]
mkdocs = "^1.4.2"
mkdocstrings = {extras = ["python"], version = "^0.21.2"}
mkdocs-material = "^9.1.6"
mkdocs-gen-files = "^0.4.0"
mkdocs-literate-nav = "^0.6.0"
ruamel-yaml = "^0.17.21"
gitpython = "^3.1.31"
[tool.poetry.extras]
timm = ["timm"]
clip = ["clip", "ftfy", "regex"]
[build-system]
requires = ["poetry-core>=1.5.0"]
build-backend = "poetry.core.masonry.api"
[tool.black]
line-length = 101
target-version = ["py38"]
[tool.isort]
profile = "black"
line_length = 101
skip_gitignore = true
remove_redundant_aliases = true