torchview/pyproject.toml at main · mert-kurttutan/torchview · 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
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
[project]
name = "torchview"
version = "0.2.7"
description = "Visualization of Pytorch Models"
authors = [{ name = "Mert Kurttutan", email = "kurttutan.mert@gmail.com" }]
readme = "README.md"
requires-python = ">=3.9"
license = { text = "MIT" }
keywords = ["pytorch", "visualization", "keras", "torch", "deep learning", "machine learning", "ml", "neural network"]
maintainers = [
    { name = "Mert Kurttutan", email = "kurttutan.mert@gmail.com" },
]
classifiers = [
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Programming Language :: Python :: 3.14",
]
dependencies = [
    "graphviz",
]

[project.urls]
homepage = "https://github.com/mert-kurttutan/torchview"
repository = "https://github.com/mert-kurttutan/torchview"

[build-system]
requires = ["hatchling>=1.22.0,<2.0.0"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["src/torchview"]

[dependency-groups]
dev = [
    "black",
    "codecov",
    "ruff==0.0.246",
    "isort",
    "pycln",
    "pylint",
    "pytest",
    "pytest-cov",
    "pre-commit",
    "packaging",
    "mkdocs",
    "mkdocs-material",
    "mkdocs-glightbox",
    "mkdocs-static-i18n",
    "mkdocs-redirects",
    "mdformat",
    "mdformat-mkdocs",
    "mkdocs-gen-files",
    "mkdocstrings",
    "mkdocstrings-python",
    "prek",
]
torch = [
    "torch",
    "torchvision",
    "torchtext",
]

[tool.setuptools.package-data]
torchview = ["py.typed"]

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]
src = ["src"]
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",]