code: def __init__(self, data_path="./data/"):
      super(MultiBodyTrainDataset, self).__init__(
          data_path + "n_body_train.npz"
      )
      self.stat_median = self.zipfile["median"]
      self.stat_max = self.zipfile["max"]
      self.stat_min = self.zipfile["min"]
  
imports: import copy
  import os
  import dgl
  import networkx
  import numpy
  import torch
  from torch.utils.data import DataLoader, Dataset
label_desc: Use image node_id 1 to create a new MultiBodyTrainDataset object from inherited base classes: MultiBodyDataset with example: obj = MultiBodyTrainDataset(data_path)