code: def _get_graph_independent(graphs_tuple):
      embed_fn = lambda x: x * 2
      out = models.GraphMapFeatures(embed_fn, embed_fn, embed_fn)(
          graphs_tuple
      )
      expected_out = graphs_tuple._replace(
          nodes=graphs_tuple.nodes * 2,
          edges=graphs_tuple.edges * 2,
          globals=graphs_tuple.globals * 2,
      )
      return out, expected_out
  
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 10 for calling a global function with example usage: _get_graph_independent(graphs_tuple) and returns: out,  expected_out