code: def parse_fn(example):
      results = tf.map_fn(
          single_parse_fn, example, dtype=(tf.uint8, tf.int64)
      )
      return tf.to_float(results[0]), results[1]
  
imports: import pytest
  from pyspark.ml.linalg import DenseVector
  from zoo.feature.common import ChainedPreprocessing, FeatureSet
  from zoo.feature.image import *
  from test.zoo.pipeline.utils.test_utils import ZooTestCase
  from zoo.pipeline.api.keras.optimizers import Adam
  from zoo.tfpark import TFNet, TFOptimizer
  import tensorflow
  import numpy
  import os
  from zoo.tfpark import KerasModel, TFDataset
label_desc: Use image node_id 2 for calling a global function with example usage: parse_fn(example) and returns: tf,  results