()
| 32 | |
| 33 | |
| 34 | def main(): |
| 35 | no_publish() |
| 36 | setup( |
| 37 | name="polygraphy", |
| 38 | version=polygraphy.__version__, |
| 39 | description="Polygraphy: A Deep Learning Inference Prototyping and Debugging Toolkit", |
| 40 | long_description=open("README.md", "r", encoding="utf-8").read(), |
| 41 | url="https://github.com/NVIDIA/TensorRT/tree/main/tools/Polygraphy", |
| 42 | author="NVIDIA", |
| 43 | author_email="svc_tensorrt@nvidia.com", |
| 44 | classifiers=[ |
| 45 | "Intended Audience :: Developers", |
| 46 | "Programming Language :: Python :: 3", |
| 47 | ], |
| 48 | license="Apache 2.0", |
| 49 | install_requires=REQUIRED_PACKAGES, |
| 50 | packages=find_packages(exclude=("tests", "tests.*")), |
| 51 | entry_points={"console_scripts": ["polygraphy=polygraphy.tools:main"]}, |
| 52 | zip_safe=True, |
| 53 | python_requires=">=3.6", |
| 54 | ) |
| 55 | |
| 56 | |
| 57 | if __name__ == "__main__": |
no test coverage detected