code: def set_stage(self, stage: Stages):
          
          assert stage in Stages
          LOG.info(f'Update session: stage {self._stage.name}->{stage.name}')
          self._stage = stage
docs: Convenience method to set the stage and lookup message