flat_structured_observations

class maze.perception.perception_utils.flat_structured_observations(structured_obs: Dict[Union[str, int], Dict[str, torch.Tensor]])

Compiles a flat dict from a structured observation nested dictionary.

Param

The structured dictionary of observations.

Returns

The flattened observation dictionary.