class_docstr: Wrapper for tf.data.Dataset to permit extensions.
class_docstr_tok: ["Wrapper","for","tf.data.Dataset","to","permit","extensions","."]
code: def __init__(
      self,
      data: tf.data.Dataset,
      image_shape: Tuple[int, int, int],
      parse_fn: Optional[Callable] = record_parse,
  ):
      self.data = data
      self.parse_fn = parse_fn
      self.image_shape = image_shape
  
imports: import os
  from typing import Callable, Optional, Tuple, List, Iterable
  import numpy
  import tensorflow
  from absl import flags
label_desc: Use image node_id 1 to create a new DataSet object  with example: obj = DataSet(data, image_shape, parse_fn)