-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
125 lines (116 loc) · 3.31 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
122
123
124
125
[tool]
[tool.poetry]
name = "aimet-ml"
version = "1.0.1"
homepage = "https://github.com/aimet-tech/aimet-ml"
description = "Python package of frequently used modules for ML developments in AIMET.."
authors = ["Pasit Jakkrawankul <[email protected]>"]
readme = "README.md"
license = "MIT"
classifiers=[
'Development Status :: 2 - Pre-Alpha',
'Intended Audience :: Developers',
'License :: OSI Approved :: MIT License',
'Natural Language :: English',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.9',
]
packages = [
{ include = "aimet_ml" },
{ include = "tests", format = "sdist" },
]
[tool.poetry.dependencies]
python = ">=3.9, <3.10"
black = { version = "^23.7", optional = true}
isort = { version = "^5.12.0", optional = true}
flake8 = { version = "^6.1.0", optional = true}
flake8-docstrings = { version = "^1.7.0", optional = true }
mypy = {version = "^1.5.1", optional = true}
pytest = { version = "^7.4.0", optional = true}
pytest-cov = { version = "^4.1.0", optional = true}
tox = { version = "^4.10.0", optional = true}
virtualenv = { version = "^20.24.3", optional = true}
pip = { version = "^23.2.1", optional = true}
mkdocs = { version = "^1.5.2", optional = true}
mkdocs-include-markdown-plugin = { version = "^6.0.0", optional = true}
mkdocs-material = { version = "^9.2.3", optional = true}
mkdocstrings = { version = "^0.22.0", optional = true}
mkdocs-material-extensions = { version = "^1.1.1", optional = true}
mkdocstrings-python = { version = "^1.5.2", optional = true}
twine = { version = "^4.0.2", optional = true}
mkdocs-autorefs = {version = "^0.5.0", optional = true}
pre-commit = {version = "^3.3.3", optional = true}
toml = {version = "^0.10.2", optional = true}
bump2version = {version = "^1.0.1", optional = true}
types-pyyaml = {version = "^6.0.12.12", optional = true}
python-dotenv = {version = "^1.0.0", optional = true}
torch = {version = "^2.0.1", optional = true}
transformers = {version = "^4.36.2", optional = true}
boto3 = "^1.28.47"
ffmpeg_python = "^0.2.0"
ffprobe-python = "^1.0.3"
static-ffmpeg = "^2.5"
matplotlib = "^3.7.2"
msgpack = "^1.0.7"
numpy = "^1.24.4"
opencv_python = "^4.8.0.76"
pandas = "^1.5.3"
pydub = "^0.25.1"
librosa = "^0.10.1"
wandb = "^0.15.10"
pyyaml = "^6.0.1"
[tool.poetry.extras]
test = [
"pytest",
"black",
"isort",
"mypy",
"flake8",
"flake8-docstrings",
"pytest-cov",
"types-pyyaml",
"python-dotenv",
]
dev = ["tox", "pre-commit", "virtualenv", "pip", "twine", "toml", "bump2version"]
doc = [
"mkdocs",
"mkdocs-include-markdown-plugin",
"mkdocs-material",
"mkdocstrings",
"mkdocs-material-extension",
"mkdocs-autorefs",
"mkdocstrings-python"
]
transformers = ["torch", "transformers"]
[tool.black]
line-length = 120
skip-string-normalization = true
target-version = ['py39']
include = '\.pyi?$'
exclude = '''
/(
\.eggs
| \.git
| \.hg
| \.mypy_cache
| \.tox
| \.venv
| _build
| buck-out
| build
| dist
)/
'''
[tool.isort]
multi_line_output = 3
include_trailing_comma = true
force_grid_wrap = 0
use_parentheses = true
ensure_newline_before_comments = true
line_length = 120
skip_gitignore = true
# you can skip files as below
#skip_glob = docs/conf.py
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"