[aws_cpi] move task_checkpoint into Helpers