obtain_normalization_statistics

class maze.core.wrappers.observation_normalization.observation_normalization_utils.obtain_normalization_statistics(env: MazeEnv | ObservationNormalizationWrapper, n_samples: int)

Obtain the normalization statistics of a given environment.

• Returns None, if the ObservationNormalizationWrapper is not implemented

• Returns the loaded statistics, if available

• Runs the estimation and returns the newly calculated statistics, if not loaded previously

Parameters:

• env – Environment with applied ObservationNormalizationWrapper (function returns None immediately if this is not the case.

• n_samples – Number of samples (=steps) to collect normalization statistics at the beginning of the training.

Returns:

The normalization statistics or None if the ObservationNormalizationWrapper is not implemented by the env.