code: def load_data(self, dataset: Dataset) -> Any:
      if self.training and hasattr(dataset, "num_classes"):
          self.num_classes = dataset.num_classes
      self.dataset = dataset
      return range(len(self.dataset))
  
file_path: lightning-flash/src/flash/pointcloud/segmentation/input.py
imports: from typing import Any
  from torch.utils.data import Dataset
  from flash.core.data.io.input import DataKeys, Input
  from flash.core.utilities.imports import requires
  from flash.pointcloud.segmentation.open3d_ml.sequences_dataset import SequencesDataset
label_desc: Use image node_id 1 for calling the PointCloudSegmentationDatasetInput obj's underlying member method code with example usage: obj.load_data(dataset) and returns: range