fix setup.py install reqs

danielenricocahall · Dec 7, 2022 · 711f04c · 711f04c
1 parent cd4bc83
commit 711f04c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -2,13 +2,17 @@
 from setuptools import find_packages
 
 setup(name='elephas',
-      version='3.4.2',
+      version='3.4.3',
       description='Deep learning on Spark with Keras',
       url='http://github.com/danielenricocahall/elephas',
       download_url='https://github.com/danielenricocahall/elephas/tarball/3.4.0',
       author='Daniel Cahall',
       author_email='danielenricocahall@gmail.com',
-      install_requires=[line.strip() for line in open('requirements.txt').readlines()],
+      iinstall_requires=['cython',
+                        'tensorflow>=2,!=2.2.*,<=2.10',
+                        'flask',
+                        'h5py==3.3.0',
+                        'pyspark<3.4'],
       extras_require={
         'tests': ['pytest', 'pytest-pep8', 'pytest-cov', 'pytest-spark', 'mock']
     },