From 43028f655d44eb524fc988a1645b993cefd08e6a Mon Sep 17 00:00:00 2001
From: dongdaxiang <dongdaxiang@baidu.com>
Date: Mon, 17 Dec 2018 00:12:03 +0800
Subject: [PATCH] refine import path for ps_instance.py test=develop

---
 python/paddle/fluid/distributed/helper.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/python/paddle/fluid/distributed/helper.py b/python/paddle/fluid/distributed/helper.py
index 999c8d77b83..cdde5403cda 100644
--- a/python/paddle/fluid/distributed/helper.py
+++ b/python/paddle/fluid/distributed/helper.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import ps_pb2 as pslib
-
 
 class FileSystem(object):
     """
@@ -37,6 +35,7 @@ class FileSystem(object):
         assert user != None
         assert passwd != None
         assert hadoop_bin != None
+        import ps_pb2 as pslib
         self.fs_client = pslib.FsClientParameter()
         #if fs_type == "afs":
         #    fs_client.fs_type = pslib.FsApiType.AFS
-- 
GitLab