code: def apply_wrapper(wrapper, task_or_dataset=None):
      if task_or_dataset is None:
          return wrapper
  
      from torchmeta.utils.data import MetaDataset
  
      if isinstance(task_or_dataset, Task):
          return wrapper(task_or_dataset)
      elif isinstance(task_or_dataset, MetaDataset):
          if task_or_dataset.dataset_transform is None:
              dataset_transform = wrapper
          else:
              dataset_transform = Compose(
                  [task_or_dataset.dataset_transform, wrapper]
              )
          task_or_dataset.dataset_transform = dataset_transform
          return task_or_dataset
      else:
          raise NotImplementedError()
  
label_desc: Use image node_id 1 for calling a global function with example usage: apply_wrapper(wrapper, task_or_dataset) and returns: wrapper, wrapper, task_or_dataset