Skip to content

Commit b820f2e

Browse files
Update setup.py
1 parent 2425ef0 commit b820f2e

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

setup.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,17 @@
11
#!/usr/bin/env python
22

3+
# Always prefer setuptools over distutils
4+
from os import path
35
from setuptools import setup
46

57
import pytorch_lightning
68

7-
# https://packaging.python.org/guides/single-sourcing-package-version/
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('#')]
813

14+
# https://packaging.python.org/guides/single-sourcing-package-version/
915
# http://blog.ionelmc.ro/2014/05/25/python-packaging/
1016
setup(
1117
name="pytorch-lightning",
@@ -25,11 +31,7 @@
2531

2632
keywords=["deep learning", "pytorch", "AI"],
2733
python_requires=">=3.6",
28-
install_requires=[
29-
"torch>=1.1.0",
30-
"tqdm",
31-
"test-tube>=0.6.7.6",
32-
],
34+
install_requires=requirements,
3335

3436
classifiers=[
3537
'Environment :: Console',

0 commit comments

Comments
 (0)