schola.core.spaces.discrete.merge_discrete_like_spaces

schola.core.spaces.discrete.merge_discrete_like_spaces(*spaces)[source]

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

Copied!

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

Copied!

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

Copied!

>>> merged_space = merge_discrete_like_spaces(DiscreteSpace(3))
>>> merged_space == DiscreteSpace(3)
True