diff --git a/paddle/fluid/framework/CMakeLists.txt b/paddle/fluid/framework/CMakeLists.txt index d6cce2540188aa3a29b97472b27b753fd5bf4bdb..361f25d09e70006e0dcba31977379a756ba8c96f 100644 --- a/paddle/fluid/framework/CMakeLists.txt +++ b/paddle/fluid/framework/CMakeLists.txt @@ -157,7 +157,7 @@ if(WITH_PYTHON) #Generate an empty \ #__init__.py to make framework_py_proto as a valid python module. add_custom_target(framework_py_proto_init ALL COMMAND ${CMAKE_COMMAND} -E touch __init__.py) - add_dependencies(framework_py_proto framework_py_proto_init) + add_dependencies(framework_py_proto framework_py_proto_init trainer_py_proto) if (NOT WIN32) add_custom_command(TARGET framework_py_proto POST_BUILD COMMAND ${CMAKE_COMMAND} -E make_directory ${PADDLE_BINARY_DIR}/python/paddle/fluid/proto diff --git a/python/paddle/fluid/trainer_desc.py b/python/paddle/fluid/trainer_desc.py index f323439c55071436989a363b7f8d1fc840b65fc1..2a206e7fa2f359d15fa28d5a44b7b7c2869f1dd1 100644 --- a/python/paddle/fluid/trainer_desc.py +++ b/python/paddle/fluid/trainer_desc.py @@ -33,15 +33,11 @@ class TrainerDesc(object): # Workaround for relative import in protobuf under python3 # TODO: should be fixed cur_path = os.path.dirname(__file__) - sys.path.append(cur_path) - sys.path.append(cur_path + "/proto") - - # Just for fix random UT fail, will be deleted soon - print("cur_path: ", cur_path) - print("sys.path: ", sys.path) - if os.path.isdir(cur_path + "/proto"): - print("[listdir]") - print(os.listdir(cur_path + "/proto")) + if cur_path not in sys.path: + sys.path.append(cur_path) + if cur_path + "/proto" not in sys.path: + sys.path.append(cur_path + "/proto") + from proto import trainer_desc_pb2 self.proto_desc = trainer_desc_pb2.TrainerDesc() import multiprocessing as mp