code: def augment_data(batch):
      return batch
  
file_path: ray/doc/source/train/doc_code/data_ingest_torch_new.py
imports: import ray
  from ray import train
  from ray.train import ScalingConfig
  from ray.train.torch import TorchTrainer
  import numpy
  from typing import Dict
  from ray.train import DataConfig
  from typing import Optional, Dict, List
  from ray.data import Dataset, DataIterator, NodeIdStr
  from ray.actor import ActorHandle
label_desc: Use image node_id 3 for calling a global function with example usage: augment_data(batch) and returns: batch