Load requirements from requirements.txt to avoid duplication and out-…

…of-sync issues
badgids · Jul 30, 2020 · 9e63cf4 · 9e63cf4
1 parent c5bda20
commit 9e63cf4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 28 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 torch>=1.5
-tensorflow>=2.2
+tensorflow==2.3.0rc0
 numpy>=1.16.0
 scipy>=0.19.0
 numba==0.48

diff --git a/setup.py b/setup.py
@@ -76,34 +76,13 @@ def pip_install(package_name):
     )
 
 
+reqs_from_file = open('requirements.txt').readlines()
+reqs_without_tf = [r for r in reqs_from_file if not r.startswith('tensorflow')]
+tf_req = [r for r in reqs_from_file if r.startswith('tensorflow')]
+
 requirements = {
-    'install_requires':[
-        "torch>=1.5",
-        "numpy>=1.16.0",
-        "numba==0.48",
-        "scipy>=0.19.0",
-        "librosa==0.7.2",
-        "unidecode==0.4.20",
-        "attrdict",
-        "tensorboardX",
-        "matplotlib",
-        "Pillow",
-        "flask",
-        "tqdm",
-        "inflect",
-        "pysbd",
-        "bokeh==1.4.0",
-        "soundfile",
-        "phonemizer>=2.2.0",
-        "nose==1.3.7",
-        "cardboardlint==1.3.0",
-        "pylint==2.5.3",
-        'fuzzywuzzy',
-        'gdown'
-    ],
-    'pip_install':[
-        'tensorflow==2.3.0rc0',
-    ]
+    'install_requires': reqs_without_tf,
+    'pip_install': tf_req
 }