def load_state_dict(self, state_dict: Dict[str, Any]) -> None:
        """
        Load the schedulers state.

        Parameters
        ----------
        state_dict : ``Dict[str, Any]``
            Scheduler state. Should be an object returned from a call to ``state_dict``.
        """
        self.__dict__.update(state_dict)