code: def main():
      run = wandb.init()
      res = Parallel(n_jobs=2)(
          delayed(f)(run, i**2) for i in range(4)
      )
      print(res)
  
imports: from math import sqrt
  import wandb
  from joblib import Parallel, delayed
label_desc: Use image node_id 2 for calling a global function with example usage: main() without return types