class_docstr: Dataset for test triplets in dict format.
  
  Args:
      triplets (dict):
          The collection of triplets with keys 'h', 'r' and 't'.
          The values are 1D np.ndarray.
      num_ents (int):
          Number of entities.
class_docstr_tok: ["Dataset","for","test","triplets","in","dict","format",".","Args",":","triplets","(","dict",")",":","The","collection","of","triplets","with","keys","'","h","'",",","'","r","'","and","'t","'",".","The","values","are","1D","np.ndarray",".","num_ents","(","int",")",":","Number","of","entities","."]
code: def __len__(self):
      return self._h.shape[0]
  
imports: import os
  import numpy
  from numpy.random import default_rng
  import paddle
  from paddle.io import Dataset
  from paddle.io import DataLoader, DistributedBatchSampler
  from utils import timer_wrapper
label_desc: Use image node_id 2 for calling the TestKGDataset obj's underlying member method code with example usage: obj.__len__() and returns: self