<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def set_stage(self, stage: Stages):
 """ Convenience method to set the stage and lookup message """

        assert stage in Stages
        LOG.info(f'Update session: stage {self._stage.name}->{stage.name}')
        self._stage = stage