code: def main():
      wandb.setup()
      num_proc = 4
      pool = mp.Pool(processes=num_proc)
      result = pool.map_async(do_run, range(num_proc))
  
      data = result.get(60)
      print(f"DEBUG: {data}")
      assert len(data) == 4
  
imports: import multiprocessing
  import wandb
  import yea
label_desc: Use image node_id 2 for calling a global function with example usage: main() without return types