4141else :
4242 long_description = 'See ' + __homepage__
4343
44-
4544# ======================= Reading Requirements files as TXT files =======================
4645
4746def req_file (filename ):
@@ -51,6 +50,24 @@ def req_file(filename):
5150 # Example: `\n` at the end of each line
5251 return [x .strip () for x in content ]
5352
53+ # ======================= Defining the requirements var =======================
54+
55+ install_requires = req_file ("requirements.txt" )
56+
57+ extras_require = {
58+ 'tf_cpu' : ['tensorflow>=1.8.0,<1.9' ],
59+ 'tf_gpu' : ['tensorflow-gpu>=1.8.0,<1.9' ],
60+ 'dev' : req_file ("requirements_dev.txt" ),
61+ 'doc' : req_file ("docs/requirements.txt" ),
62+ 'test' : req_file ("tests/requirements.txt" )
63+ }
64+
65+ # Readthedocs requires TF 1.5.0 to build properly
66+ if os .environ .get ('READTHEDOCS' , None ) == 'True' :
67+ install_requires .append ("tensorflow==1.5.0" )
68+
69+ # ======================= Define the package setup =======================
70+
5471setup (
5572 name = __package_name__ ,
5673
@@ -125,16 +142,12 @@ def req_file(filename):
125142 # your project is installed. For an analysis of "install_requires" vs pip's
126143 # requirements files see:
127144 # https://packaging.python.org/en/latest/requirements.html
128- install_requires = req_file ( "requirements.txt" ) ,
145+ install_requires = install_requires ,
129146
130147 # List additional groups of dependencies here (e.g. development
131148 # dependencies). You can install these using the following syntax,
132149 # $ pip install -e .[test]
133- extras_require = {
134- 'dev' : req_file ("requirements_dev.txt" ),
135- 'doc' : req_file ("docs/requirements.txt" ),
136- 'test' : req_file ("tests/requirements.txt" )
137- },
150+ extras_require = extras_require ,
138151 scripts = [
139152 'tl' ,
140153 ],
0 commit comments