code: def get_dataset(self, stage: str = "train"):
      dataloader = getattr(self.datamodule, f"{stage}_dataloader")()
      dataset = dataloader.dataset.dataset
      if isinstance(dataset, TorchDataloader):
          return dataset.dataset
      return dataset
  
file_path: lightning-flash/src/flash/pointcloud/segmentation/open3d_ml/app.py
imports: import torch
  from flash.core.data.data_module import DataModule
  from flash.core.data.io.input import DataKeys
  from flash.core.utilities.imports import _TOPIC_POINTCLOUD_AVAILABLE
label_desc: Use image node_id 2 for calling the App obj's underlying member method code with example usage: obj.get_dataset(stage) and returns: dataset, dataset