UnSqueezePreProcessor

class maze.core.wrappers.observation_preprocessing.preprocessors.unsqueeze.UnSqueezePreProcessor(observation_space: gymnasium.spaces.Box, dim: int)

An un-squeeze pre-processor.

Parameters:
  • observation_space – The observation space to pre-process.

  • dim – Index where to add an additional dimension.

process(observation: numpy.ndarray) numpy.ndarray

(overrides PreProcessor)

implementation of PreProcessor interface

processed_shape() Tuple[int, ...]

(overrides PreProcessor)

implementation of PreProcessor interface

processed_space() gymnasium.spaces.Box

(overrides PreProcessor)

implementation of PreProcessor interface