code: def setup_method(self, method):
      tf.keras.backend.clear_session()
      super(TestTFDataset, self).setup_method(method)
  
imports: import pytest
  from pyspark.ml.linalg import DenseVector
  from bigdl.dllib.feature.common import ChainedPreprocessing, FeatureSet
  from bigdl.dllib.feature.image import *
  from bigdl.orca.test_zoo_utils import ZooTestCase
  from bigdl.dllib.keras.optimizers import Adam
  from bigdl.orca.tfpark import TFNet, TFOptimizer
  import tensorflow
  import numpy
  import os
  from bigdl.orca.tfpark import KerasModel, TFDataset
label_desc: Use image node_id 2 for calling the TestTFDataset obj's underlying member method code with example usage: obj.setup_method(method) without return types