code: def get_dataset(size, config):
      data = get_linear_data(size=size)
      dataset = tf.data.Dataset.from_tensor_slices(data)
      dataset = dataset.batch(config["batch_size"])
      return dataset
  
file_path: BigDL/python/orca/test/bigdl/orca/automl/model/test_base_keras_model.py
imports: from unittest import TestCase
  from bigdl.orca.automl.model.base_keras_model import KerasBaseModel, KerasModelBuilder
  import numpy
  import tensorflow
  import pytest
label_desc: Use image node_id 2 for calling a global function with example usage: get_dataset(size, config) and returns: dataset