fix loading pkg while setup (#71)

* fix loading package while setup #56

* Update setup.py

* Update __init__.py

* Update setup.py

* use complete req.

* Update setup.py

* Update setup.py
This commit is contained in:
Jirka Borovec 2019-08-08 11:17:23 +02:00 committed by William Falcon
parent 514cab131e
commit d0ccbb8e1c
2 changed files with 26 additions and 35 deletions

View File

@ -2,16 +2,6 @@ from .models.trainer import Trainer
from .root_module.root_module import LightningModule from .root_module.root_module import LightningModule
from .root_module.decorators import data_loader from .root_module.decorators import data_loader
__version__ = '0.3.6.9'
__author__ = "William Falcon",
__author_email__ = "waf2107@columbia.edu"
__license__ = 'Apache-2'
__homepage__ = 'https://github.com/williamFalcon/pytorch-lightning',
__copyright__ = 'Copyright (c) 2018-2019, %s.' % __author__
__doc__ = """
The Keras for ML researchers using PyTorch
"""
__all__ = [ __all__ = [
'Trainer', 'Trainer',
'LightningModule', 'LightningModule',

View File

@ -1,38 +1,39 @@
#!/usr/bin/env python #!/usr/bin/env python
# Always prefer setuptools over distutils # Always prefer setuptools over distutils
from os import path from setuptools import setup, find_packages
from setuptools import setup
import pytorch_lightning
PATH_HERE = path.abspath(path.dirname(__file__))
with open(path.join(PATH_HERE, 'requirements.txt'), encoding='utf-8') as fp:
requirements = [rq.rstrip() for rq in fp.readlines() if not rq.startswith('#')]
# https://packaging.python.org/guides/single-sourcing-package-version/ # https://packaging.python.org/guides/single-sourcing-package-version/
# http://blog.ionelmc.ro/2014/05/25/python-packaging/ # http://blog.ionelmc.ro/2014/05/25/python-packaging/
# 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...
# the goal of the project is simplicity for researchers, don't want to add too much
# engineer specific practices
setup( setup(
name="pytorch-lightning", name='pytorch-lightning',
version=pytorch_lightning.__version__, version='0.3.6.9',
description=pytorch_lightning.__doc__, description='The Keras for ML researchers using PyTorch',
author=pytorch_lightning.__author__, author='William Falcon',
author_email=pytorch_lightning.__author_email__, author_email='waf2107@columbia.edu',
url=pytorch_lightning.__homepage__, url='https://github.com/williamFalcon/pytorch-lightning',
license=pytorch_lightning.__license__, download_url='https://github.com/williamFalcon/pytorch-lightning',
packages=['pytorch_lightning'], license='Apache-2',
packages=find_packages(),
long_description=open("README.md", encoding="utf-8").read(), long_description=open('README.md', encoding='utf-8').read(),
long_description_content_type='text/markdown', long_description_content_type='text/markdown',
include_package_data=True, include_package_data=True,
zip_safe=False, zip_safe=False,
keywords=['deep learning', 'pytorch', 'AI'],
keywords=["deep learning", "pytorch", "AI"], python_requires='>=3.6',
python_requires=">=3.6", install_requires=[
install_requires=requirements, 'torch>=1.1.0',
'tqdm',
'test-tube>=0.6.7.6',
'pandas>=0.20.3',
],
classifiers=[ classifiers=[
'Environment :: Console', 'Environment :: Console',
'Natural Language :: English', 'Natural Language :: English',