class_docstr: Wrapper for tf.data.Dataset to permit extensions.
class_docstr_tok: ["Wrapper","for","tf.data.Dataset","to","permit","extensions","."]
code: def augment(self, para_augment: int = 4):
      if self.augment_fn:
          return self.map(self.augment_fn, para_augment)
      return self
  
file_path: objax/examples/image_classification/tfdata/data.py
imports: from typing import Callable, Optional, Tuple, List
  import numpy
  import tensorflow
label_desc: Use image node_id 7 for calling the DataSet obj's underlying member method code with example usage: obj.augment(para_augment) and returns: self, self