code: def train_func():
      for i in range(3):
          train.report(dict(epoch=i))
  
imports: from pathlib import Path
  from ray import train
  from ray.train import RunConfig, ScalingConfig
  from ray.train.torch import TorchTrainer
  from ray.tune.logger import TBXLoggerCallback
  from ray.tune.logger.mlflow import MLflowLoggerCallback
label_desc: Use image node_id 1 for calling a global function with example usage: train_func() without return types