add a wrap_decorator decorator to keep the __doc__ attr (#45773)

add a wrap_decorator decorator to keep the doc attr (#45773)
fd86a938 · Nyakku Shigure · GitHub · d4c4c53d · fd86a938
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/fluid/dygraph/jit.py python/paddle/fluid/dygraph/jit.py +1 -0

未找到文件。
--- a/python/paddle/fluid/dygraph/jit.py
+++ b/python/paddle/fluid/dygraph/jit.py
@@ -634,6 +634,7 @@ def _remove_save_pre_hook(hook):
    _save_pre_hooks_lock.release()


+@wrap_decorator
 def _run_save_pre_hooks(func):

    def wrapper(layer, path, input_spec=None, **configs):