code: def sample(self, edge_idx):
      if not isinstance(edge_idx, torch.Tensor):
          edge_idx = torch.tensor(edge_idx)
      row, col, _ = self.adj_t.coo()
      batch = torch.cat([row[edge_idx], col[edge_idx]], dim=0)
      return super(PositiveLinkNeighborSampler, self).sample(batch)
  
imports: import argparse
  from tqdm.auto import tqdm
  import torch
  import torch.nn.functional
  from torch.utils.data import DataLoader
  from torch_geometric.data import NeighborSampler
  from torch_geometric.nn import SAGEConv
  from torch_geometric.utils import to_undirected
  from ogb.linkproppred import PygLinkPropPredDataset, Evaluator
  from logger import Logger
label_desc: Use image node_id 2 for calling the PositiveLinkNeighborSampler obj's underlying member method code with example usage: obj.sample(edge_idx) and returns: super