code: def set_control_scheme(self, index):
          
          self._current_control_scheme = index % self._num_control_schemes
          self._control_scheme_buffer[0] = self._current_control_scheme
docs: 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`.