code: def train_fn(config):
      for _i in range(10):
          loss = config["a"] + config["b"]
          wandb.log({"loss": loss})
          tune.report(loss=loss)
  
imports: import random
  import numpy
  import wandb
  from _test_support import get_wandb_api_key
  from ray import tune
  from ray.tune.integration.wandb import wandb_mixin
label_desc: Use image node_id 1 for calling a global function with example usage: train_fn(config) without return types