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:
parent
514cab131e
commit
d0ccbb8e1c
|
@ -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',
|
||||||
|
|
51
setup.py
51
setup.py
|
@ -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',
|
||||||
|
|
Loading…
Reference in New Issue