def set_state(self, state):
        if self._state != state:
            self._state = state
            if self.dag_id is not None:
                # something really weird goes on here: if you try to close the session