setup.cfg

[metadata]
name = bitorchinfo
version = attr: bitorchinfo.__version__
description = Model summary for bitorch, based on the original torchinfo.
long_description = file: README.md
long_description_content_type = text/markdown
url = https://github.com/hpi-xnor/bittorchinfo
author = Joseph Bethge
author_email = joseph.bethge@hpi.de
license = MIT
license_file = LICENSE
classifiers =
    License :: OSI Approved :: MIT License
    Operating System :: OS Independent
    Programming Language :: Python :: 3
    Programming Language :: Python :: 3 :: Only
    Programming Language :: Python :: 3.6
    Programming Language :: Python :: 3.7
    Programming Language :: Python :: 3.8
    Programming Language :: Python :: 3.9
    Programming Language :: Python :: 3.10
keywords = bitorch bitorchinfo torch pytorch torchsummary torch-summary summary keras deep-learning ml torchinfo torch-info visualize model statistics layer stats

[options]
packages = bitorchinfo
python_requires = >=3.6
include_package_data = True

[options.package_data]
torchinfo = py.typed

[mypy]
strict = True
implicit_reexport = True
allow_untyped_calls = True
show_error_codes = True

[pylint.MASTER]
load-plugins = pylint_strict_informational

[pylint.MESSAGES CONTROL]
extension-pkg-whitelist = torch
enable =
    useless-suppression,
    deprecated-pragma,
    use-symbolic-message-instead,
disable =
    missing-function-docstring,
    too-many-instance-attributes,
    too-many-arguments,
    too-many-branches,
    too-many-locals,
    invalid-name,
    no-member,
    fixme,
    duplicate-code,

[isort]
profile = black

[flake8]
max-line-length = 88
extend-ignore = E203,F401