code: def maybe_expand(img, axis=2):
      if len(img.shape) != axis:
          return img
      return np.expand_dims(img, axis=axis)
  
imports: import itertools
  import os
  import numpy
  from PIL import Image, ImageEnhance, ImageOps
  from nose2.tools import params, cartesian_params
  import nvidia.dali.tensors
  from nvidia.dali import fn, pipeline_def
  from nvidia.dali.auto_aug import augmentations
  from nvidia.dali.auto_aug.core._utils import get_translations
  from test_utils import get_dali_extra_path, check_batch
label_desc: Use image node_id 2 for calling a global function with example usage: maybe_expand(img, axis) and returns: np, img