[aws_cpi] revert task_checkpoint changes in resource_wait