code: def gen(n_body, num_steps, orbit):
      # initialization on just first state
      data = init(num_steps, n_body, fea_num, orbit)
      for i in range(1, num_steps):
          data[i] = calc(data[i - 1], n_body)
      return data
  
file_path: dgl/examples/pytorch/graphsim/n_body_sim.py
imports: from __future__ import absolute_import, division, print_function
  import argparse
  import os
  from math import cos, pi, radians, sin
  import numpy
label_desc: Use image node_id 6 for calling a global function with example usage: gen(n_body, num_steps, orbit) and returns: data