diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index 4b5961ac2d..235590f292 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -1,3 +1,13 @@ from .models.trainer import Trainer from .root_module.root_module import LightningModule -from .root_module.decorators import data_loader \ No newline at end of file +from .root_module.decorators import data_loader + +__version__ = '0.3.6.9' +__author__ = "William Falcon", +__author_email__ = "waf2107@columbia.edu" +__license__ = 'MIT' +__homepage__ = 'https://github.com/williamFalcon/pytorch-lightning', +__copyright__ = 'Copyright (c) 2018-2019, %s.' % __author__ +__doc__ = """ +The Keras for ML researchers using PyTorch +""" diff --git a/setup.py b/setup.py index e680ef4301..5f076a36d5 100755 --- a/setup.py +++ b/setup.py @@ -1,19 +1,28 @@ #!/usr/bin/env python -from setuptools import setup, find_packages +from setuptools import setup + +import pytorch_lightning # https://packaging.python.org/guides/single-sourcing-package-version/ # http://blog.ionelmc.ro/2014/05/25/python-packaging/ setup( name="pytorch-lightning", - version='0.3.6.9', - description="The Keras for ML researchers using PyTorch", - author="William Falcon", - author_email="waf2107@columbia.edu", - url="https://github.com/williamFalcon/pytorch-lightning", - download_url="https://github.com/williamFalcon/pytorch-lightning", - license="MIT", + version=pytorch_lightning.__version__, + description=pytorch_lightning.__doc__, + author=pytorch_lightning.__author__, + author_email=pytorch_lightning.__author_email__, + url=pytorch_lightning.__homepage__, + license=pytorch_lightning.__license__, + packages=['pytorch_lightning'], + + long_description=open("README.md", encoding="utf-8").read(), + long_description_content_type='text/markdown', + + include_package_data=True, + zip_safe=False, + keywords=["deep learning", "pytorch", "AI"], python_requires=">=3.5", install_requires=[ @@ -21,9 +30,26 @@ setup( "tqdm", "test-tube>=0.6.7.6", ], - packages=find_packages(), - long_description=open("README.md", encoding="utf-8").read(), - long_description_content_type='text/markdown', - include_package_data=True, - zip_safe=False, + + classifiers=[ + 'Environment :: Console', + 'Natural Language :: English', + # How mature is this project? Common values are + # 3 - Alpha, 4 - Beta, 5 - Production/Stable + 'Development Status :: 4 - Beta', + # Indicate who your project is intended for + 'Intended Audience :: Developers', + 'Topic :: Scientific/Engineering :: Artificial Intelligence', + 'Topic :: Scientific/Engineering :: Image Recognition', + 'Topic :: Scientific/Engineering :: Information Analysis', + # Pick your license as you wish + 'License :: OSI Approved :: BSD License', + 'Operating System :: OS Independent', + # Specify the Python versions you support here. In particular, ensure + # that you indicate whether you support Python 2, Python 3 or both. + 'Programming Language :: Python :: 3', + 'Programming Language :: Python :: 3.5', + 'Programming Language :: Python :: 3.6', + 'Programming Language :: Python :: 3.7', + ], )