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