class_docstr: A learner wrapper that normalizes observations using running statistics.
class_docstr_tok: ["A","learner","wrapper","that","normalizes","observations","using","running","statistics","."]
code: def restore(self, state: NormalizationLearnerWrapperState):
      self._wrapped_learner.restore(state.wrapped_learner_state)
      self._observation_running_statistics = (
          state.observation_running_statistics
      )
  
imports: import dataclasses
  import functools
  from typing import Any, Callable, Generic, Iterator, List, Optional, Tuple
  import acme
  from acme import adders
  from acme import core
  from acme import specs
  from acme import types
  from acme.agents.jax import builders
  from acme.jax import networks
  from acme.jax import running_statistics
  from acme.jax import variable_utils
  from acme.jax.types import Networks, Policy
  from acme.utils import counting
  from acme.utils import loggers
  import dm_env
  import jax
  import reverb
  from typing_extensions import Protocol
label_desc: Use image node_id 6 for calling the NormalizationLearnerWrapper obj's underlying member method code with example usage: obj.restore(state) without return types