Skip to content

schola.core.spaces.discrete.merge_discrete_like_spaces

schola.core.spaces.discrete.merge_discrete_like_spaces(*spaces) : Merge multiple Discrete or MultiDiscrete spaces into a single MultiDiscrete space.

Parameters: : *spaces (List*[Union[gymnasium.spaces.Discrete,* gymnasium.spaces.MultiDiscrete]**]) – The spaces to merge.

Returns: : The merged space.

Return type: : MultiDiscreteSpace

Raises: : TypeError – If any of the spaces are not Discrete or MultiDiscrete.

Examples

>>> merged_space = merge_discrete_like_spaces(MultiDiscreteSpace([3,2]), DiscreteSpace(2))
>>> merged_space == MultiDiscreteSpace([3, 2, 2])
True
>>> merged_space = merge_discrete_like_spaces(DiscreteSpace(3), DiscreteSpace(2))
>>> merged_space == MultiDiscreteSpace([3, 2])
True
>>> merged_space = merge_discrete_like_spaces(DiscreteSpace(3))
>>> merged_space == DiscreteSpace(3)
True