add a wrap_decorator decorator to keep the __doc__ attr (PaddlePaddle…

…#45773)
Caozhou1995 · Sep 8, 2022 · a865fef · a865fef
1 parent e3beb80
commit a865fef
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):