code: def f(run, x):
      # with wandb.init() as run:
      run.config.x = x
      run.define_metric(f"step_{x}")
      for i in range(3):
          # Log metrics with wandb
          run.log({f"i_{x}": i * x, f"step_{x}": i})
      return sqrt(x)
  
imports: from math import sqrt
  import wandb
  from joblib import Parallel, delayed
label_desc: Use image node_id 1 for calling a global function with example usage: f(run, x) and returns: sqrt