|
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 | + author_email='waf2107@columbia.edu', |
| 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