add PY3 requires

6ef9bab0 · tangwei · 71e03371 · 6ef9bab0
隐藏空白更改
内联并排

Showing with 13 addition and 6 deletion

setup.py setup.py +13 -6

未找到文件。
--- a/setup.py
+++ b/setup.py
 """
 setup for fleet-rec.
 """
-
+import sys
 from setuptools import setup, find_packages

-requires = [
-    "paddlepaddle >= 0.0.0",
-    "netron >= 0.0.0",
-    "yaml"
-]
+if sys.version_info.major == 2:
+    requires = [
+        "paddlepaddle >= 0.0.0",
+        "netron >= 0.0.0",
+        "yaml"
+    ]
+else:
+    requires = [
+        "paddlepaddle >= 0.0.0",
+        "netron >= 0.0.0",
+        "pyyaml"
+    ]

 about = {}
 about["__title__"] = "fleet-rec"