Cached the task_args in task_arg_gen