code: def main(args):
      if args.mode == "valid":
          val_hr = np.load(
              "./dataset/wikikg90m_kddcup2021/processed/val_hr.npy",
              mmap_mode="r",
          )[: args.nums]
      else:
          val_hr = np.load(
              "./dataset/wikikg90m_kddcup2021/processed/test_hr.npy",
              mmap_mode="r",
          )[: args.nums]
      print(args.tmp_path)
  
      scores = get_top10(val_hr)
      if args.output_path is not None:
          np.save(args.output_path, scores)
  
file_path: PGL/examples/kddcup2021/WikiKG90M/post_smoothing/get_score.py
imports: import numpy
  from glob import glob
  from tqdm import tqdm
  import sys
  import argparse
  import logging
  from multiprocessing import Pool
label_desc: Use image node_id 6 for calling a global function with example usage: main(args) without return types