Skip to content

Commit 3e4fc69

Browse files
mohi7solankiatodorov
authored andcommitted
Bugfix: Use functools.wrap to preserve leave_module info
Signed-off-by: Mohit Solanki <mohitsolanki619@gmail.com>
1 parent e43c7d7 commit 3e4fc69

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pylint_django/augmentations/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
"""Augmentations."""
22
# pylint: disable=invalid-name
3+
import functools
34
import itertools
45

56
from astroid import InferenceError
@@ -739,6 +740,7 @@ def is_class(class_name):
739740

740741

741742
def wrap(orig_method, with_method):
743+
@functools.wraps(orig_method)
742744
def wrap_func(*args, **kwargs):
743745
with_method(orig_method, *args, **kwargs)
744746
return wrap_func

0 commit comments

Comments
 (0)