code: def __call__(
      self,
      params: Params,
      observation: Observation,
      *args,
      is_training: bool,
      key: Optional[PRNGKey] = None,
      **kwargs,
  ) -> NetworkOutput:
      ...
  
file_path: acme/acme/jax/networks/base.py
imports: import dataclasses
  from typing import Callable, Optional, Tuple
  from acme import specs
  from acme import types
  from acme.jax import types
  from acme.jax import utils
  import haiku
  import jax.numpy
  from typing_extensions import Protocol
label_desc: Use image node_id 1 for calling the ApplyFn obj's underlying member method code with example usage: obj.__call__(params, observation) without return types