code: def train_fn(config):
      for i in range(100):
          ray.train.report(dict(timesteps_total=i))
  
imports: import inspect
  import unittest
  from unittest.mock import patch
  import ray
  import ray.train
  from ray.tune import register_trainable, run_experiments, run, choice
  from ray.tune.result import TIMESTEPS_TOTAL
  from ray.tune.experiment import Experiment
  from ray.tune.search.hyperopt import HyperOptSearch
  from ray.tune.experiment import Trial
  from ray.util.client.ray_client_helpers import ray_start_client_server
label_desc: Use image node_id 1 for calling a global function with example usage: train_fn(config) without return types