code: def find_similar_paper(start_pid):
      all_edges = []
      for pid in range(
          start_pid,
          min(graph_paper2paper.num_nodes, start_pid + batch_size),
      ):
          edges = find_similar_paper_single(pid)
          if len(edges) > 0:
              all_edges.append(edges)
      return np.concatenate(all_edges, dtype="int64")
  
file_path: PGL/examples/kddcup2021/MAG240M/r_unimp/construct_coauthor_graph.py
imports: import pgl
  import numpy
  from collections import defaultdict, Counter
  import multiprocessing
  import tqdm
  import time
  import pickle
  import numpy
  import numpy
  from collections import defaultdict, Counter
  import multiprocessing
  import time
  import pickle
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: find_similar_paper(start_pid) and returns: np