Load requirements from requirements.txt to avoid duplication and out-of-sync issues

This commit is contained in:
Reuben Morais 2020-07-27 14:29:14 +02:00
parent c5bda20990
commit 9e63cf4072
2 changed files with 7 additions and 28 deletions

View File

@ -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

View File

@ -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
}