forked from mert-kurttutan/torchview
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
115 lines (102 loc) · 2.65 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
[project]
name = "torchview"
version = "0.2.6"
description = "Visualization of Pytorch Models"
readme = "README.md"
requires-python = ">=3.7"
license = {'text'= "MIT"}
keywords = [
"pytorch", "visualization", "keras",
"torch", "deep learning", "machine learning",
"ml", "neural network"
]
authors = [
{email = "[email protected]"},
{name = "Mert Kurttutan"}
]
maintainers = [
{name = "Mert Kurttutan", email = "[email protected]"}
]
classifiers = [
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
]
dependencies = [
]
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"
[project.urls]
homepage = "https://github.com/mert-kurttutan/torchview"
repository = "https://github.com/mert-kurttutan/torchview"
[tool.setuptools]
# Scan the project directory with the default parameters
packages = [
'torchview',
'torchview.computation_node'
]
[tool.setuptools.package-data]
torchview = ["py.typed"]
[tool.mypy]
strict = true
implicit_reexport = true
show_error_codes = true
enable_error_code = 'ignore-without-code'
allow_untyped_calls = true
[[tool.mypy.overrides]]
module = [
'graphviz',
'torchtext',
'torchvision',
'torchvision.models',
'tests.fixtures.models',
'tests.fixtures.u_net',
'tests.fixtures.dense_net',
'tests.fixtures.custom_attention',
'torchtext.functional',
'transformers',
]
ignore_missing_imports = true
[tool.pylint."MESSAGES CONTROL"]
extension-pkg-whitelist = "torch"
enable = [
"useless-suppression",
"deprecated-pragma",
"use-symbolic-message-instead",
]
disable = [
"super-with-arguments", # explicit is better
"missing-module-docstring",
"missing-function-docstring",
"too-many-instance-attributes",
"too-many-arguments",
"too-many-branches",
"too-many-locals",
"invalid-name",
"line-too-long", # Covered by flake8
"no-member",
"fixme",
"duplicate-code",
"no-else-return", # more explicit is better
]
[tool.ruff]
select = [
"E", # pycodestyle errors
"W", # pycodestyle warnings
"F", # pyflakes
"C", # flake8-comprehensions
"B", # flake8-bugbear
]
ignore = [
"B008", # do not perform function calls in argument defaults
"C901", # too complex
"B905", # should work python < 3.10
"B010", # no problem with using setattr ???
]
[tool.ruff.per-file-ignores]
"tests/fixtures/ldc.py" = ["E402",]