schola.core.spaces.discrete.merge_discrete_like_spaces

Function Definition

schola.core.spaces.discrete.merge_discrete_like_spaces(*spaces)

Merge multiple Discrete or MultiDiscrete spaces into a single MultiDiscrete space.

Parameters

*spaces

Type: List[Union[gymnasium.spaces.Discrete, gymnasium.spaces.MultiDiscrete]]
The spaces to merge.

Returns

Type: MultiDiscreteSpace
The merged space.

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