code: def format_output(invocation: MethodInvocation) -> str:
      return pprint.pformat(invocation.output_spec)
  
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 9 for calling a global function with example usage: format_output(invocation) and returns: pprint