code: def set_epoch(self, epoch, batch_size):
      # TODO ray: implement dynamic batch size
      self.step = 0
  
imports: from ludwig.data.batcher.base import Batcher
label_desc: Use image node_id 4 for calling the IterableBatcher obj's underlying member method code with example usage: obj.set_epoch(epoch, batch_size) without return types