-
Notifications
You must be signed in to change notification settings - Fork 41
/
Copy pathpyproject.toml
121 lines (107 loc) · 3.1 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.poetry]
name = "perceiver-io"
version = "0.11.1"
description = "Perceiver IO"
readme = "README.md"
authors = [
"Martin Krasser <[email protected]>",
"Christoph Stumpf <[email protected]>",
]
license = "Apache-2.0"
keywords= ["perceiver-io", "perceiver-ar", "deep-learning"]
classifiers = [
"Environment :: Console",
"Natural Language :: English",
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
]
homepage = "https://github.com/krasserm/perceiver-io"
packages = [
{ include = "perceiver" },
]
include = ["docs"]
[tool.poetry.dependencies]
python = "^3.8,<3.11"
pytorch-lightning = "^2.0"
torch = "^2.0"
fairscale = "^0.4"
torchmetrics = "^0.9"
torch-optimizer = "^0.3"
tensorboard = "^2.11"
einops = "^0.4"
cchardet = "^2.1"
datasets = {version = "^2.4", optional = true}
tokenizers = {version = "^0.12", optional = true}
transformers = {version = "^4.28", optional = true}
torchvision = {version = "^0.15", optional = true}
opencv-python = {version = "^4.6.0.66", optional = true}
jsonargparse = {extras = ["signatures"], version = "^4.18"}
fsspec = {extras = ["s3"], version = "*"}
pretty_midi = {version = "^0.2.10", optional = true}
[tool.poetry.group.dev.dependencies]
invoke = "^1.6.0"
pre-commit = "^2.17.0"
pytest = "^7.0.1"
pytest-cov = "^3.0.0"
flaky = "^3.7.0"
[tool.poetry.group.examples]
optional = true
[tool.poetry.group.examples.dependencies]
scipy = "^1.10"
pandas = "^1.1"
notebook = "*"
matplotlib = "*"
ipywidgets = "*"
termcolor = "2.0.1"
[tool.poetry.extras]
text = ["datasets", "tokenizers", "transformers"]
vision = ["datasets", "torchvision", "opencv-python"]
audio = ["datasets", "pretty_midi", "transformers"]
[tool.black]
line-length = 120
exclude = "(.eggs|.git|.hg|.mypy_cache|.venv|_build|buck-out|build|dist)"
[tool.isort]
known_first_party = [
"perceiver",
]
skip_glob = []
profile = "black"
line_length = 120
[tool.pytest.ini_options]
[tool.coverage.run]
source = ["perceiver"]
[tool.ruff]
line-length = 120
# Enable Pyflakes `E` and `F` codes by default.
select = [
"E", "W", # see: https://pypi.org/project/pycodestyle
"F", # see: https://pypi.org/project/pyflakes
]
#extend-select = [
# "C4", # see: https://pypi.org/project/flake8-comprehensions
# "SIM", # see: https://pypi.org/project/flake8-simplify
# "RET", # see: https://pypi.org/project/flake8-return
# "PT", # see: https://pypi.org/project/flake8-pytest-style
#]
ignore = [
"E731", # Do not assign a lambda expression, use a def
]
# Exclude a variety of commonly ignored directories.
exclude = [
".git",
"docs",
]
ignore-init-module-imports = true
[tool.ruff.per-file-ignores]
"perceiver/**/__init__.py" = ["F401", "F403"]
[tool.ruff.mccabe]
# Unlike Flake8, default to a complexity level of 10.
max-complexity = 10