class_docstr: Wrapping network def for a specified return_to_condition.
code: def apply(self, params, x, support, key=None):
      del key
      return self.network_def.apply(
          params,
          x,
          support=support,
          return_to_condition=self._return_to_condition,
      )
  
imports: import functools
  from absl import logging
  from dopamine.jax import losses
  from dopamine.jax.agents.dqn import dqn_agent
  from dopamine.jax.agents.rainbow import rainbow_agent
  from dopamine.labs.offline_rl.jax import offline_rainbow_agent
  import gin
  import jax
  import jax.numpy
  import numpy
  import optax
  import tensorflow
label_desc: Use image node_id 3 for calling the WrappedNetworkDef obj's underlying member method code with example usage: obj.apply(params, x, support, key) and returns: self