code: def get_dataset(size, config):
      data = get_x_y(size=size)
      dataset = tf.data.Dataset.from_tensor_slices(data)
      dataset = dataset.batch(config["batch_size"])
      return dataset
  
imports: import tensorflow
  import numpy
  from unittest import TestCase
  from bigdl.orca.automl.auto_estimator import AutoEstimator
  import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: get_dataset(size, config) and returns: dataset