add

m3dev · Nov 21, 2023 · d3c0ef6 · d3c0ef6
1 parent 5f37e57
commit d3c0ef6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gokart/task_complete_check.py b/gokart/task_complete_check.py
@@ -1,3 +1,4 @@
+import functools
 from logging import getLogger
 from typing import Callable
 
@@ -6,6 +7,7 @@
 
 def task_complete_check_wrapper(run_func: Callable, complete_check_func: Callable):
 
+    @functools.wraps(run_func)
     def wrapper(*args, **kwargs):
         if complete_check_func():
             logger.warning(f'{run_func.__name__} is skipped because the task is already completed.')