From 87c1dc2a1a2770d95103bfb51e93897d306f39c5 Mon Sep 17 00:00:00 2001
From: Andreas Leitherer <leitherer@fhi-berlin.mpg.de>
Date: Tue, 5 Jan 2021 12:04:43 +0100
Subject: [PATCH] Modified setup.py file for nn regression

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b6cea0d..264d91d 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==1.14.0', 'keras==2.3.0', 'numpy==1.16.4', 'scipy', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'],
+    install_requires=['tensorflow==1.13.1', 'keras==2.2.4', 'keras-vis', 'numpy==1.16.4', 'scipy==1.3.1', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'],
 )
-- 
GitLab