code: def __call__(self, x: jax.Array):
      return self.wrapped(x, is_training=True)
  
imports: from collections.abc import Sequence
  from typing import Any, Callable, NamedTuple
  import haiku
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the Training obj's underlying member method code with example usage: obj.__call__(x) and returns: self