未验证 提交 db272ca5 编写于 作者: L liym27 提交者: GitHub

rename file to according to reviews. (#26236)

上级 b7a86e92
......@@ -369,13 +369,14 @@ def ast_to_func(ast_root, dyfunc, delete_on_exit=True):
function, the other inner functions are invisible for the decorated function.
"""
def remove_file(filepath):
def remove_if_exit(filepath):
if os.path.exists(filepath):
os.remove(filepath)
source = ast_to_source_code(ast_root)
import_fluid = "import paddle.fluid as fluid\n"
source = import_fluid + source
if six.PY2:
source = source.encode('utf-8')
f = tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False)
......@@ -387,8 +388,8 @@ def ast_to_func(ast_root, dyfunc, delete_on_exit=True):
f.write(source)
if delete_on_exit:
atexit.register(lambda: remove_file(f.name))
atexit.register(lambda: remove_file(f.name[:-3] + ".pyc"))
atexit.register(lambda: remove_if_exit(f.name))
atexit.register(lambda: remove_if_exit(f.name[:-3] + ".pyc"))
module = imp.load_source(module_name, f.name)
func_name = dyfunc.__name__
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册