code: def get_raw_data(self, stage, idx, **kwargs):
      assert stage in self.available_dataset_names
      dataset = self.get_dataset(stage)
      if hasattr(dataset, "get_raw_data"):
          return dataset.get_raw_data(idx, **kwargs)
      else:
          raise Exception(
              "Dataset {} doesn t have a get_raw_data function implemented".format(
                  dataset
              )
          )
  
imports: import os
  from abc import ABC, abstractmethod
  import logging
  import functools
  from functools import partial
  import numpy
  import torch
  import torch_geometric
  from torch_geometric.transforms import Compose, FixedPoints
  import copy
  from torch_points3d.models import model_interface
  from torch_points3d.core.data_transform import instantiate_transforms, MultiScaleTransform
  from torch_points3d.core.data_transform import instantiate_filters
  from torch_points3d.datasets.batch import SimpleBatch
  from torch_points3d.datasets.multiscale_data import MultiScaleBatch
  from torch_points3d.utils.enums import ConvolutionFormat
  from torch_points3d.utils.config import ConvolutionFormatFactory
  from torch_points3d.utils.colors import COLORS, colored_print
label_desc: Use image node_id 28 for calling the BaseDataset obj's underlying member method code with example usage: obj.get_raw_data(stage, idx) and returns: dataset