Merge pull request #450 from autonomio/lock_package_versions

Lock package versions for deps to legacy version
autonomio · Jan 25, 2020 · 375a5a1 · 375a5a1
2 parents 0635968 + 8e390f7
commit 375a5a1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -24,10 +24,12 @@
 except ImportError:
     from distutils.core import setup
 
-install_requires = ['wrangle',
+install_requires = ['statsmodels>=0.11.0',
+                    'wrangle',
                     'numpy',
                     'pandas',
-                    'keras',
+                    'tensorflow<=1.5.0',
+                    'keras==2.3.0',
                     'astetik',
                     'sklearn',
                     'tqdm',