From 06b326b681e2d9a831e02a7f2f0a1e8c18c1bdeb Mon Sep 17 00:00:00 2001
From: typhoonzero <typhoonzero1986@gmail.com>
Date: Fri, 12 Jan 2018 17:07:11 +0800
Subject: [PATCH] follow comments

---
 python/paddle/v2/fluid/__init__.py                     | 3 ++-
 python/paddle/v2/fluid/distribute_transpiler_simple.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/python/paddle/v2/fluid/__init__.py b/python/paddle/v2/fluid/__init__.py
index 5e01b871980..a14422ee92f 100644
--- a/python/paddle/v2/fluid/__init__.py
+++ b/python/paddle/v2/fluid/__init__.py
@@ -18,13 +18,14 @@ from param_attr import ParamAttr
 from data_feeder import DataFeeder
 from core import LoDTensor, CPUPlace, CUDAPlace
 from distribute_transpiler import DistributeTranspiler
+from distribute_transpiler_simple import SimpleDistributeTranspiler
 import clip
 
 Tensor = LoDTensor
 __all__ = framework.__all__ + executor.__all__ + [
     'io', 'initializer', 'layers', 'nets', 'optimizer', 'backward',
     'regularizer', 'LoDTensor', 'CPUPlace', 'CUDAPlace', 'Tensor', 'ParamAttr'
-    'DataFeeder', 'clip', 'DistributeTranspiler'
+    'DataFeeder', 'clip', 'SimpleDistributeTranspiler', 'DistributeTranspiler'
 ]
 
 
diff --git a/python/paddle/v2/fluid/distribute_transpiler_simple.py b/python/paddle/v2/fluid/distribute_transpiler_simple.py
index 49ece7b725e..32db3df9aa2 100644
--- a/python/paddle/v2/fluid/distribute_transpiler_simple.py
+++ b/python/paddle/v2/fluid/distribute_transpiler_simple.py
@@ -48,7 +48,7 @@ def round_robin(params_grads, pserver_endpoints):
     return param_grad_map
 
 
-class DistributeTranspiler:
+class SimpleDistributeTranspiler:
     def transpile(self,
                   optimize_ops,
                   params_grads,
-- 
GitLab