diff --git a/setup.py b/setup.py index 6f33a246ec3115898717ace17296da170014841c..b6cea0d056236db8865d688fca081c88aaf7b68b 100644 --- a/setup.py +++ b/setup.py @@ -13,5 +13,5 @@ setup( description=metainfo['title'], long_description=metainfo['description'], packages=find_packages(), - install_requires=['tensorflow<2.0', 'keras', 'numpy', 'scipy', 'matplotlib', 'pandas', 'seaborn', 'pymatgen', 'sklearn'], + install_requires=['tensorflow==1.14.0', 'keras==2.3.0', 'numpy==1.16.4', 'scipy', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'], )