code: def collate_test(self, samples):
      batch = torch.LongTensor(samples)
      blocks = self.sampler.sample_blocks(batch)
      assign_features_to_blocks(
          blocks, self.g, self.textset, self.ntype
      )
      return blocks
  
imports: import argparse
  import pickle
  import time
  import dgl
  import dgl.function
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.utils.data import DataLoader, IterableDataset
  from ..None import utils
label_desc: Use image node_id 3 for calling the PinSAGECollator obj's underlying member method code with example usage: obj.collate_test(samples) and returns: blocks