|
1 | 1 | #!/usr/bin/env python |
2 | 2 |
|
3 | 3 | # Always prefer setuptools over distutils |
4 | | -from os import path |
5 | | -from setuptools import setup |
6 | | - |
7 | | -import pytorch_lightning |
8 | | - |
9 | | -PATH_HERE = path.abspath(path.dirname(__file__)) |
10 | | - |
11 | | -with open(path.join(PATH_HERE, 'requirements.txt'), encoding='utf-8') as fp: |
12 | | - requirements = [rq.rstrip() for rq in fp.readlines() if not rq.startswith('#')] |
| 4 | +from setuptools import setup, find_packages |
13 | 5 |
|
14 | 6 | # https://packaging.python.org/guides/single-sourcing-package-version/ |
| 7 | + |
15 | 8 | # http://blog.ionelmc.ro/2014/05/25/python-packaging/ |
16 | | -setup( |
17 | | - name="pytorch-lightning", |
18 | | - version=pytorch_lightning.__version__, |
19 | | - description=pytorch_lightning.__doc__, |
20 | | - author=pytorch_lightning.__author__, |
21 | | - author_email=pytorch_lightning.__author_email__, |
22 | | - url=pytorch_lightning.__homepage__, |
23 | | - license=pytorch_lightning.__license__, |
24 | | - packages=['pytorch_lightning'], |
25 | 9 |
|
26 | | - long_description=open("README.md", encoding="utf-8").read(), |
| 10 | +# https://packaging.python.org/discussions/install-requires-vs-requirements/ |
| 11 | +# keep the meta-data here for simplicity in reading this file... it's not obvious |
| 12 | +# what happens and to non-engineers they won't know to look in init... |
| 13 | +# the goal of the project is simplicity for researchers, don't want to add too much |
| 14 | +# engineer specific practices |
| 15 | +setup( |
| 16 | + name='pytorch-lightning', |
| 17 | + version='0.3.6.9', |
| 18 | + description='The Keras for ML researchers using PyTorch', |
| 19 | + author='William Falcon', |
| 20 | + |
| 21 | + url='https://github.com/williamFalcon/pytorch-lightning', |
| 22 | + download_url='https://github.com/williamFalcon/pytorch-lightning', |
| 23 | + license='Apache-2', |
| 24 | + packages=find_packages(), |
| 25 | + long_description=open('README.md', encoding='utf-8').read(), |
27 | 26 | long_description_content_type='text/markdown', |
28 | | - |
29 | 27 | include_package_data=True, |
30 | 28 | zip_safe=False, |
31 | | - |
32 | | - keywords=["deep learning", "pytorch", "AI"], |
33 | | - python_requires=">=3.6", |
34 | | - install_requires=requirements, |
35 | | - |
| 29 | + keywords=['deep learning', 'pytorch', 'AI'], |
| 30 | + python_requires='>=3.6', |
| 31 | + install_requires=[ |
| 32 | + 'torch>=1.1.0', |
| 33 | + 'tqdm', |
| 34 | + 'test-tube>=0.6.7.6', |
| 35 | + 'pandas>=0.20.3', |
| 36 | + ], |
36 | 37 | classifiers=[ |
37 | 38 | 'Environment :: Console', |
38 | 39 | 'Natural Language :: English', |
|
0 commit comments