code: def _get_graph_network_no_edge_update(graphs_tuple):
      # Our test update functions are just identity functions.
      update_node_fn = lambda n, se, re, g: n
      update_edge_fn = None
      update_global_fn = lambda gn, ge, g: g
      net = models.GraphNetwork(
          update_edge_fn, update_node_fn, update_global_fn
      )
      return net(graphs_tuple)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  from jraph._src import models
  from jraph._src import utils
  import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: _get_graph_network_no_edge_update(graphs_tuple) and returns: net