diff --git a/gokart/task_complete_check.py b/gokart/task_complete_check.py index 4ab8a2b5..cf072a6b 100644 --- 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.')