diff --git a/setup.py b/setup.py index 3806700..780869e 100644 --- a/setup.py +++ b/setup.py @@ -1,11 +1,12 @@ from setuptools import setup, find_packages setup( - name = "torchsummaryX", - version = "1.1.0", - description = "Improved visualization tool of torchsummary.", - author = "Namhyuk Ahn", - author_email = "nmhkahn@gmail.com", - url = "https://github.com/nmhkahn/torchsummaryX", - packages =["torchsummaryX"], + name = "torchsummaryX", + version = "1.2.0", + description = "Improved visualization tool of torchsummary.", + author = "Namhyuk Ahn", + author_email = "nmhkahn@gmail.com", + url = "https://github.com/nmhkahn/torchsummaryX", + packages =["torchsummaryX"], + install_requires = ["torch", "numpy", "pandas"], )