code: def data(self):
      import logging
  
      logger = logging.getLogger("PyGCuGraphTrainer")
      logger.info("getting data")
  
      if self.__data is None:
          # FIXME wholegraph
          fs = FeatureStore(backend="torch")
          num_nodes_dict = {}
  
          for node_type, x in self.__dataset.x_dict.items():
              logger.debug(f"getting x for {node_type}")
              fs.add_data(x, node_type, "x")
              num_nodes_dict[node_type] = self.__dataset.num_nodes(
                  node_type
              )
  
          for node_type, y in self.__dataset.y_dict.items():
              logger.debug(f"getting y for {node_type}")
              fs.add_data(y, node_type, "y")
  
          for node_type, train in self.__dataset.train_dict.items():
              logger.debug(f"getting train for {node_type}")
              fs.add_data(train, node_type, "train")
  
          for node_type, test in self.__dataset.test_dict.items():
              logger.debug(f"getting test for {node_type}")
              fs.add_data(test, node_type, "test")
  
          for node_type, val in self.__dataset.val_dict.items():
              logger.debug(f"getting val for {node_type}")
              fs.add_data(val, node_type, "val")
  
          # TODO support online sampling if the edge index is provided
          num_edges_dict = self.__dataset.edge_index_dict
          if not isinstance(list(num_edges_dict.values())[0], int):
              num_edges_dict = {k: len(v) for k, v in num_edges_dict}
  
          self.__data = CuGraphStore(
              fs,
              num_edges_dict,
              num_nodes_dict,
          )
  
      logger.info(f"got data successfully on rank {self.rank}")
  
      return self.__data
  
imports: from .trainers_pyg import PyGTrainer
  from models.pyg import CuGraphSAGE
  import torch
  import numpy
  from torch.nn.parallel import DistributedDataParallel
  from cugraph.gnn import FeatureStore
  from cugraph_pyg.data import CuGraphStore
  from cugraph_pyg.loader import BulkSampleLoader
  import os
label_desc: Use image node_id 8 for calling the PyGCuGraphTrainer obj's underlying member method code with example usage: obj.data() and returns: self