code: def batch_dataset(dataset, batch_size):
      if batched(dataset):
          return dataset
      return dataset.batch(batch_size)
  
file_path: autokeras/autokeras/utils/data_utils.py
imports: import numpy
  import tensorflow
  from tensorflow import nest
label_desc: Use image node_id 2 for calling a global function with example usage: batch_dataset(dataset, batch_size) and returns: dataset, dataset