code: def __init__(self, p, dists, num_layers):
      super().__init__()
      self.p = p
      self.dists = dists
      self.num_layers = num_layers
  
imports: import dgl
  import dgl.function
  import numpy
  import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new CARESampler object from inherited base classes: dgl.dataloading with example: obj = CARESampler(p, dists, num_layers)