code: def __init__(
      self,
      datamodule: pl.LightningDataModule,
      generator: Optional[torch.Generator] = None,
  ):
      super().__init__()
      self.datamodule = datamodule
      self.generator = generator
  
file_path: opacus/opacus/lightning.py
imports: from typing import Any, Optional
  import pytorch_lightning
  import torch
  from opacus.data_loader import DPDataLoader
label_desc: Use image node_id 1 to create a new DPLightningDataModule object from inherited base classes: pl with example: obj = DPLightningDataModule(datamodule, generator)