class_docstr: DeepChem Lightning Dataset Module to be used with the DCLightningModule and a Lightning trainer.
  
  This module wraps over the the deepchem pytorch dataset and dataloader providing a generic interface to run training.
  https://pytorch-lightning.readthedocs.io/en/stable/extensions/datamodules.html
  
  Notes
  -----
  This class requires PyTorch to be installed.
code: def setup(self, stage):
      self.train_dataset = self._dataset
  
file_path: deepchem/deepchem/models/lightning/dc_lightning_dataset_module.py
imports: from typing import Callable
  import deepchem
  import lightning
  import torch
label_desc: Use image node_id 2 for calling the DCLightningDatasetModule obj's underlying member method code with example usage: obj.setup(stage) without return types