code: def __init__(self, dset):
      super().__init__()
      self.dset = dset
  
file_path: generative-models/sgm/data/cifar10.py
imports: import pytorch_lightning
  import torchvision
  from torch.utils.data import DataLoader, Dataset
  from torchvision import transforms
label_desc: Use image node_id 1 to create a new CIFAR10DataDictWrapper object from inherited base classes: Dataset with example: obj = CIFAR10DataDictWrapper(dset)