SpacesConfig

class maze.perception.models.space_config.SpacesConfig(action_spaces_dict: Dict[str | int, gymnasium.spaces.Dict], observation_spaces_dict: Dict[str | int, gymnasium.spaces.Dict], agent_counts_dict: Dict[str | int, int])

Represents configuration of environment spaces (action & observation) used for model config.

Spaces config are needed (together with model config and dumped state dict) when loading a trained policy for rollout.

classmethod load(in_file_path: str) → SpacesConfig

Load a saved spaces config from a file.

Parameters:

in_file_path – Where to load the spaces config from.

Returns:

Loaded spaces config object

save(dump_file_path: str) → None

Save the spaces config to a file.

Parameters:

dump_file_path – Where to save the spaces config.