From 6ef9bab0cc036c62a88b76390313a5c5b57d4760 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Wed, 29 Apr 2020 14:04:42 +0800
Subject: [PATCH] add PY3 requires

---
 setup.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/setup.py b/setup.py
index bdf0f05a..14e3b6bd 100644
--- a/setup.py
+++ b/setup.py
@@ -1,14 +1,21 @@
 """
 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"
-- 
GitLab