code: def format_input(invocation: MethodInvocation) -> str:
      out = ", ".join(map(pprint.pformat, invocation.args_spec))
      kwargs = invocation.kwargs_spec
      if kwargs:
          out += "\n"
          out += ", ".join(
              f"{k}={pprint.pformat(v)}" for k, v in kwargs.items()
          )
      return out
  
imports: from collections.abc import Mapping, Sequence
  import dataclasses
  import functools
  import pprint
  from typing import Any, Callable, Optional, TypeVar, Union
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import module
  from haiku._src import transform
  from haiku._src import utils
  import jax
  import jax.numpy
  import numpy
  import tabulate
label_desc: Use image node_id 8 for calling a global function with example usage: format_input(invocation) and returns: out