code: def solarize_add_shift(shift):
      if shift >= 128:
          raise Exception(
              "The solarize_add augmentation accepts shifts from 0 to 128"
          )
      return np.uint8(shift)
  
imports: from nvidia.dali import fn
  from nvidia.dali import types
  from nvidia.dali.auto_aug.core import augmentation
label_desc: Use image node_id 20 for calling a global function with example usage: solarize_add_shift(shift) and returns: np