[aws_cpi] Move task_checkpoint to helpers