From a355f4f0fcd56df48a39c58ae5541c4c2600fd92 Mon Sep 17 00:00:00 2001 From: Gunnlaugur Thor Briem Date: Thu, 5 Mar 2020 11:11:42 +0000 Subject: [PATCH] Add functools.wraps for wrapper initializer Preserve the original initializer function's metadata. See https://docs.python.org/3/library/functools.html#functools.update_wrapper --- src/transformers/modeling_tf_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py index 122ee9c911..29049f7371 100644 --- a/src/transformers/modeling_tf_utils.py +++ b/src/transformers/modeling_tf_utils.py @@ -14,8 +14,7 @@ # See the License for the specific language governing permissions and # limitations under the License. """TF general model utils.""" - - +import functools import logging import os @@ -54,6 +53,7 @@ def keras_serializable(cls): if config_class is None: raise AttributeError("Must set `config_class` to use @keras_serializable") + @functools.wraps(initializer) def wrapped_init(self, config, *args, **kwargs): if isinstance(config, dict): config = config_class.from_dict(config)