From 4db9d890f1bd981e5166b948dad5410eddf5c3c8 Mon Sep 17 00:00:00 2001
From: liweibin <liweibin02@baidu.com>
Date: Fri, 25 Oct 2019 15:26:04 +0800
Subject: [PATCH] update

---
 docs/requirements.txt |  1 -
 requirements.txt      |  2 +-
 setup.py              | 13 +++++--------
 3 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index 8178df6..4e7960b 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -3,6 +3,5 @@ mistune
 sphinx_rtd_theme
 numpy >= 1.16.4
 cython >= 0.25.2
-networkx
 paddlepaddle
 pgl
diff --git a/requirements.txt b/requirements.txt
index e0cabc0..4a13b16 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,4 @@ cython >= 0.25.2
 
 #paddlepaddle
 
-redis-py-cluster == 1.3.6
+redis-py-cluster
diff --git a/setup.py b/setup.py
index f0b808a..78833f5 100755
--- a/setup.py
+++ b/setup.py
@@ -24,16 +24,14 @@ from setuptools.command.build_ext import build_ext as _build_ext
 try:
     from Cython.Build import cythonize
 except ImportError:
-
-    def cythonize(*args, **kwargs):
-        """cythonize"""
-        from Cython.Build import cythonize
-        return cythonize(*args, **kwargs)
+     def cythonize(*args, **kwargs):
+         """cythonize"""
+         from Cython.Build import cythonize
+         return cythonize(*args, **kwargs)
 
 
 class CustomBuildExt(_build_ext):
     """CustomBuildExt"""
-
     def finalize_options(self):
         _build_ext.finalize_options(self)
         # Prevent numpy from thinking it is still in its setup process:
@@ -41,7 +39,6 @@ class CustomBuildExt(_build_ext):
         import numpy
         self.include_dirs.append(numpy.get_include())
 
-
 workdir = os.path.dirname(os.path.abspath(__file__))
 with open(os.path.join(workdir, './requirements.txt')) as f:
     requirements = f.read().splitlines()
@@ -111,7 +108,7 @@ setup(
         'numpy>=1.16.4',
     ],
     install_requires=requirements,
-    cmdclass={'build_ext': CustomBuildExt},
+    cmdclass={'build_ext':CustomBuildExt},
     packages=find_packages(),
     include_package_data=True,
     #ext_modules=cythonize(extensions),
-- 
GitLab