code: def mnist_dataset(c: Configs):
      """
      Create MNIST dataset
      """
      return MNISTDataset(c.image_size)
  
imports: from typing import List
  import torch
  import torch.utils.data
  import torchvision
  from PIL import Image
  from labml import lab, tracker, experiment, monit
  from labml.configs import BaseConfigs, option
  from labml_helpers.device import DeviceConfigs
  from labml_nn.diffusion.ddpm import DenoiseDiffusion
  from labml_nn.diffusion.ddpm.unet import UNet
label_desc: Use image node_id 2 for calling a global function with example usage: mnist_dataset(c) and returns: MNISTDataset