diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index 7bdaf7b558..533322c8f5 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -7,3 +7,13 @@ __all__ = [ 'LightningModule', 'data_loader', ] + +__version__ = '0.5.2.1' +__author__ = ' William Falcon et al.' +__author_email__ = 'waf2107@columbia.edu' +__license__ = 'Apache-2.0' +__homepage__ = 'https://github.com/williamFalcon/pytorch-lightning', +__doc__ = """# PyTorch Lightning + +The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate. +""" diff --git a/setup.py b/setup.py index 883e71cd21..869fed186d 100755 --- a/setup.py +++ b/setup.py @@ -7,6 +7,8 @@ from setuptools import setup, find_packages # http://blog.ionelmc.ro/2014/05/25/python-packaging/ +import pytorch_lightning + # https://packaging.python.org/discussions/install-requires-vs-requirements / # keep the meta-data here for simplicity in reading this file... it's not obvious # what happens and to non-engineers they won't know to look in init ... @@ -14,13 +16,13 @@ from setuptools import setup, find_packages # engineer specific practices setup( name='pytorch-lightning', - version='0.5.2.1', - description='The Keras for ML researchers using PyTorch', - author='William Falcon', - author_email='waf2107@columbia.edu', - url='https://github.com/williamFalcon/pytorch-lightning', + version=pytorch_lightning.__version__, + description=pytorch_lightning.__doc__, + author=pytorch_lightning.__author__, + author_email=pytorch_lightning.__author_email__, + url=pytorch_lightning.__homepage__, download_url='https://github.com/williamFalcon/pytorch-lightning', - license='Apache-2', + license=pytorch_lightning.__license__, packages=find_packages(), long_description=open('README.md', encoding='utf-8').read(), long_description_content_type='text/markdown',