def get_input_space(cls):
        """
        Define the input space as a Dict Space
        Returns: Dict action space

        """
        action_space = super(ExtraEnvInputPolicy, cls).get_input_space()
        return gym.spaces.Dict({"action": action_space, "extra": cls.extra_input_space})

        Define the input space as a Dict Space
        Returns: Dict action space

        
metadrive/policy/env_input_policy.py