code: def _make_nest(array):
      """Returns a nest given an array."""
      return {
          "a": array,
          "b": [jnp.ones_like(array), {"c": jnp.zeros_like(array)}],
      }
  
imports: import functools
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax.lib import xla_bridge
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  from jraph._src import utils
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: _make_nest(array) and returns: dict