code: def test_InferDataset():
      config_file = "../../../config.yaml"
      ip_list_file = "../../../ip_list.txt"
      config = prepare_config(config_file)
  
      ds = InferDataset(config, ip_list_file)
      loader = Dataloader(ds, batch_size=1, num_workers=1)
      for data in loader:
          print(data[0])
          break
  
imports: import os
  import sys
  import time
  import warnings
  import numpy
  from collections import defaultdict
  from pgl.graph_kernel import skip_gram_gen_pair
  from pgl.utils.logger import log
  from pgl.distributed import DistGraphClient, DistGraphServer
  from pgl.utils.data import Dataloader, StreamDataset
  from utils.config import prepare_config
  from datasets.node import NodeGenerator
  from datasets.walk import WalkGenerator
  from datasets.pair import PairGenerator
label_desc: Use image node_id 2 for calling a global function with example usage: test_InferDataset() without return types