code: def main():
      def z(shape):
          return np.zeros(shape, dtype=np.uint8)
  
      seq = iaa.Identity()
  
      print("This should generate NO warning:")
      _image_aug = seq.augment_images(z((1, 16, 16, 3)))
  
      print("This should generate NO warning:")
      _image_aug = seq.augment_images(z((16, 16, 8)))
  
      print("This should generate NO warning:")
      _image_aug = seq.augment_images([z((16, 16, 3))])
  
      print("This should generate NO warning:")
      _image_aug = seq.augment_images([z((16, 16))])
  
      print("This should generate a warning:")
      _image_aug = seq.augment_images(z((16, 16, 3)))
  
      print("This should generate a warning:")
      for _ in range(2):
          _image_aug = seq.augment_images(z((16, 16, 1)))
  
      print("This should fail:")
      _image_aug = seq.augment_images(z((16, 16)))
  
label_desc: Use image node_id 1 for calling a global function with example usage: main() and returns: np