make_normalized_env_factory

class maze.core.wrappers.observation_normalization.observation_normalization_utils.make_normalized_env_factory(env_factory: Callable[], Union[maze.core.env.structured_env.StructuredEnv, maze.core.env.structured_env_spaces_mixin.StructuredEnvSpacesMixin, maze.core.wrappers.observation_normalization.observation_normalization_wrapper.ObservationNormalizationWrapper]], normalization_statistics: Dict[str, Dict[str, Union[numpy.ndarray, float, int, Iterable[Union[float, int]]]]])

Wrap an existing env factory to assign the passed normalization statistics.

Parameters

  • env_factory – The existing env factory

  • normalization_statistics – The normalization statistics that should be applied to the env

Returns

The wrapped env factory