-

cool-RR · cool-RR · commit fd9a5f4e72f5 · 2019-07-18T19:29:28.000+03:00
diff --git a/python_toolbox/cute_profile/cute_profile.py b/python_toolbox/cute_profile/cute_profile.py
@@ -85,7 +85,8 @@ def f(x, y):
 
     def decorator(function):
 
-        def inner(function_, *args, **kwargs):
+        @functools.wraps(function)
+        def inner(*args, **kwargs):
 
             if decorated_function.condition is not None:
 
@@ -120,7 +121,7 @@ def inner(function_, *args, **kwargs):
 
                 return decorated_function.original_function(*args, **kwargs)
 
-        decorated_function = decorator(inner, function)
+        decorated_function = inner
 
         decorated_function.original_function = function
         decorated_function.profiling_on = None
