make_normalized_env_factory

class maze.core.wrappers.observation_normalization.observation_normalization_utils.make_normalized_env_factory(env_factory: Callable[[], StructuredEnv | StructuredEnvSpacesMixin | ObservationNormalizationWrapper], normalization_statistics: Dict[str, Dict[str, numpy.ndarray | float | int | Iterable[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