File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change 11#!/usr/bin/env python
22
3+ # Always prefer setuptools over distutils
4+ from os import path
35from setuptools import setup
46
57import 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/
1016setup (
1117 name = "pytorch-lightning" ,
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' ,
You can’t perform that action at this time.
0 commit comments