fix: h5py version in python>=3.9 (#500)

* fix: h5py version in python>=3.9 Co-authored-by: 浅梦 <[email protected]>
shenweichen · Nov 9, 2022 · e756480 · e756480
1 parent 91dc7c6
commit e756480
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/setup.py b/setup.py
@@ -5,14 +5,11 @@
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-if sys.version_info < (3, 9):
-    REQUIRED_PACKAGES = [
-        'h5py==2.10.0', 'requests'
-    ]
-else:
-    REQUIRED_PACKAGES = [
-        'h5py==3.7.0', 'requests'
-    ]
+REQUIRED_PACKAGES = [
+    'requests',
+    'h5py==3.7.0; python_version>="3.9"',
+    'h5py==2.10.0; python_version<"3.9"'
+]
 
 setuptools.setup(
     name="deepctr",