To punish or to leave: Distinct cognitive processes underlie partner control and partner choice behaviors