async def set_state(self, token: str, state: BaseState):
        """Set the state for a token.

        Args:
            token: The token to set the state for.
            state: The state to set.
        """
Set the state for a token.

        Args:
            token: The token to set the state for.
            state: The state to set.
        
reflex/istate/manager.py