code: def estimate_normalization_statistics(
      env: ObservationNormalizationWrapper,
  ) -> ObservationNormalizationWrapper:
      """Estimates observation normalization statistics.
      :param env: A observation normalization wrapped structured environment.
      :return: The given env with initialized normalization parameters.
      """
      env.set_observation_collection(status=True)
      env.reset()
      for _ in range(100):
          action = env.action_space.sample()
          _, _, done, _ = env.step(action)
  
          if done:
              break
  
      env.estimate_statistics()
      env.set_observation_collection(status=False)
  
      return env
  
file_path: maze/maze/test/shared_test_utils/observation_normalization.py
imports: from functools import partial
  from typing import Dict, Callable, Iterable
  import gym
  import numpy
  from maze.core.env.maze_env import CoreEnvType
  from maze.core.env.structured_env import StructuredEnv
  from maze.core.utils.factory import ConfigType
  from maze.core.wrappers.observation_normalization.observation_normalization_wrapper import ObservationNormalizationWrapper
label_desc: Use image node_id 1 for calling a global function with example usage: estimate_normalization_statistics(env) and returns: env