code: def copy_structure(bundle: T) -> T:
      return jax.tree_util.tree_map(lambda x: x, bundle)
  
imports: import collections
  import collections.abc
  from collections.abc import Mapping, MutableMapping
  import functools
  import inspect
  from typing import Any, Callable, Optional, TypeVar
  from haiku._src import base
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling a global function with example usage: copy_structure(bundle) and returns: jax