code: def add_graphs_tuples(
      graphs: jraph.GraphsTuple, other_graphs: jraph.GraphsTuple
  ) -> jraph.GraphsTuple:
      """Adds the nodes, edges and global features from other_graphs to graphs."""
      return graphs._replace(
          nodes=graphs.nodes + other_graphs.nodes,
          edges=graphs.edges + other_graphs.edges,
          globals=graphs.globals + other_graphs.globals,
      )
  
file_path: flax/examples/ogbg_molpcba/models.py
imports: from typing import Callable, Sequence
  from flax import linen
  import jax.numpy
  import jraph
label_desc: Use image node_id 1 for calling a global function with example usage: add_graphs_tuples(graphs, other_graphs) and returns: graphs