code: def get_static_graph(graph: jraph.GraphsTuple) -> jraph.GraphsTuple:
      """Returns the graph with the static parts of a system only."""
      nodes = dict(graph.nodes)
      del nodes["position"], nodes["momentum"]
      return graph._replace(nodes=frozendict(nodes))
  
imports: import functools
  from typing import Tuple, Callable
  from absl import app
  from frozendict import frozendict
  import jax
  import jax.numpy
  import jraph
  import matplotlib.pyplot
  import numpy
label_desc: Use image node_id 7 for calling a global function with example usage: get_static_graph(graph) and returns: graph