--

cool-RR · cool-RR · commit 72f49b20cb5c · 2019-07-18T19:38:15.000+03:00
diff --git a/python_toolbox/cute_profile/cute_profile.py b/python_toolbox/cute_profile/cute_profile.py
@@ -11,7 +11,7 @@
 import marshal
 
 from python_toolbox import misc_tools
-from python_toolbox.third_party.decorator import decorator
+from python_toolbox.third_party.decorator import decorator as decorator_
 
 from . import base_profile
 from . import profile_handling
@@ -85,8 +85,7 @@ def f(x, y):
 
     def decorator(function):
 
-        @functools.wraps(function)
-        def inner(*args, **kwargs):
+        def inner(function, *args, **kwargs):
 
             if decorated_function.condition is not None:
 
@@ -121,7 +120,7 @@ def inner(*args, **kwargs):
 
                 return decorated_function.original_function(*args, **kwargs)
 
-        decorated_function = inner
+        decorated_function = decorator_(inner, function)
 
         decorated_function.original_function = function
         decorated_function.profiling_on = None
