SlotCurri/pyproject.toml at main · wjun0830/SlotCurri · 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
73
74
75
[tool.poetry]
name = "slotcurry"
version = "0.1.0"
description = "SlotCurri"
authors = [
  "WonJun Moon",
  "Hyun Seok Seong",
  "Jae-Pil Heo"
]
readme = "README.md"

[tool.poetry.dependencies]
python = "~3.10"
numpy = "^1.23.3"
torch = "^1.13.0"
torchvision = "^0.14.0"
torchmetrics = "^1.2.0"
pytorch-lightning = "^1.8.3"
timm = "^0.9.7"
omegaconf = "^2.2.3"
webdataset = "^0.2.35"
einops = "^0.6.0"
scipy = "^1.10.1"
tensorflow-datasets = {version = "^4.7.0", optional=true}
tensorflow-cpu = {version = "^2.11.0", optional=true}
notebook = {version = "^6.5.2", optional=true}
matplotlib = {version = "^3.6.2", optional=true}
moviepy = "^1.0.3"
pycocotools = {version = "^2.0.6", optional=true}
requests = "^2.31.0"
av = "^12.0.0"

[tool.poetry.extras]
tensorflow = ["tensorflow-cpu", "tensorflow_datasets"]
coco = ["pycocotools"]
notebook = ["notebook", "matplotlib"]

[tool.poetry.group.dev.dependencies]
black = "^22.10.0"
isort = "^5.12.0"
pytest = "^7.1.3"
pre-commit = "^2.20.0"
ruff = "^0.0.292"

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

[tool.black]
line-length = 101
target-version = ["py38"]

[tool.isort]
profile = "black"
line_length = 101
skip_gitignore = true

[tool.ruff]
ignore = ["E731"]
line-length = 101
select = [
  "B",
  "E",
  "F",
  "TID",
  "W",
]

[tool.pytest.ini_options]
filterwarnings = "ignore::DeprecationWarning"
markers = [
  "slow: Marks slow tests",
  "dataset: Marks tests with dataset dependencies"
]
addopts = " -m 'not slow'"  # Do not run those tests by default