code: def __init__(self, data_dir: str = "./"):
      super().__init__()
      self.data_dir = data_dir
      self.non_picklable = None
      self.checkpoint_state: Optional[str] = None
  
imports: from typing import Optional
  import torch
  from pytorch_lightning import LightningDataModule, LightningModule
  from torch.utils.data import DataLoader, Dataset, Subset
label_desc: Use image node_id 1 to create a new BoringDataModule object from inherited base classes: LightningDataModule with example: obj = BoringDataModule(data_dir)