code: def collate_dataset_wrapper(batch):
      return DCLightningDatasetBatch(batch)
  
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 1 for calling a global function with example usage: collate_dataset_wrapper(batch) and returns: DCLightningDatasetBatch