code: def __getitem__(self, index):
      if self._ms_transform:
          return MultiScaleBatch.from_data_list(self.datalist)
      else:
          return Batch.from_data_list(self.datalist)
  
imports: import torch
  from torch_geometric.data import Data, Batch
  from torch_points3d.datasets.batch import SimpleBatch
  from torch_points3d.core.data_transform import MultiScaleTransform
  from torch_points3d.datasets.multiscale_data import MultiScaleBatch
label_desc: Use image node_id 1 for calling the MockDatasetGeometric obj's underlying member method code with example usage: obj.__getitem__(index) and returns: MultiScaleBatch, Batch