def set_control_scheme(self, index):
        """Sets the control scheme for the agent. See :obj:`ControlSchemes`.

        Args:
            index (int): The control scheme to use. Should be set with an enum from :obj:`ControlSchemes`.
        """
        self._current_control_scheme = index % self._num_control_schemes
        self._control_scheme_buffer[0] = self._current_control_scheme