code: def _format_shape(var):
      return var.aval.str_short().replace("float", "f")
  
imports: from collections.abc import Mapping, Sequence
  import dataclasses
  import itertools
  import logging
  import os
  import sys
  from typing import Any, Callable, NamedTuple, Optional
  from haiku._src import summarise
  import jax
  import jax.core
label_desc: Use image node_id 3 for calling a global function with example usage: _format_shape(var) and returns: var