code: def test_estimator_graph_tf_dataset_v2(self):
      def dataset_creator():
          dataset = tf.data.Dataset.from_tensor_slices(
              (
                  np.random.randint(0, 200, size=(100,)),
                  np.random.randint(0, 50, size=(100,)),
                  np.ones(shape=(100,), dtype=np.int32),
              )
          )
          return dataset._dataset
  
      self._test_estimator_graph_tf_dataset(dataset_creator)
  
imports: import os
  import shutil
  import tempfile
  from unittest import TestCase
  import numpy
  import tensorflow
  from bigdl.orca.learn.trigger import SeveralIteration
  from pyspark.sql.context import SQLContext
  import bigdl.orca.data.pandas
  from bigdl.dllib.nncontext import init_nncontext
  from bigdl.orca import OrcaContext
  from bigdl.orca.data.tf.data import Dataset
  from bigdl.orca.learn.tf.estimator import Estimator
  from bigdl.dllib.utils.tf import save_tf_checkpoint, load_tf_checkpoint, get_checkpoint_state
label_desc: Use image node_id 16 for calling the TestEstimatorForGraph obj's underlying member method code with example usage: obj.test_estimator_graph_tf_dataset_v2() and returns: dataset