code: def created_drifted_collate_function(collate_fn):
      def drifted_collate_function(batch):
          data_dict = collate_fn(batch)
          data_dict["images"] = [
              add_brightness(np.array(img))
              for img in data_dict["images"]
          ]
          return data_dict
  
      return drifted_collate_function
  
imports: import numpy
  from deepchecks.vision.checks import ImageDatasetDrift
  from deepchecks.vision.datasets.detection.coco_torch import load_dataset
label_desc: Use image node_id 2 for calling a global function with example usage: created_drifted_collate_function(collate_fn) and returns: drifted_collate_function, data_dict