def set_extra_input_space(cls, extra_input_space: gym.spaces.space.Space):
        """
        Set the space for this extra input. Error will be thrown, if this class property is set already.
        Args:
            extra_input_space: gym.spaces.space.Space

        Returns: None

        """
        assert isinstance(extra_input_space, gym.spaces.space.Space)
        ExtraEnvInputPolicy.extra_input_space = extra_input_space

        Set the space for this extra input. Error will be thrown, if this class property is set already.
        Args:
            extra_input_space: gym.spaces.space.Space

        Returns: None

        
metadrive/policy/env_input_policy.py