code: def _create_lookup_service(
      partitions_dir, ntypes, id_map, rank, world_size, num_parts
  ):
      id_lookup = dist_lookup.DistLookupService(
          partitions_dir, ntypes, rank, world_size, num_parts
      )
      id_lookup.set_idMap(id_map)
  
      # invoke the main function here.
      print(
          f"[Rank: {rank}] Done with Dist Lookup Service initialization..."
      )
  
      return id_lookup
  
imports: import logging
  import os
  import platform
  import tempfile
  from datetime import timedelta
  import dgl
  import numpy
  import pyarrow
  import pytest
  import torch.distributed
  import torch.multiprocessing
  from pytest_utils import create_chunked_dataset
  from tools.distpartitioning import constants, dist_lookup
  from tools.distpartitioning.gloo_wrapper import allgather_sizes
  from tools.distpartitioning.utils import get_idranges, get_ntype_counts_map, read_json
label_desc: Use image node_id 2 for calling a global function with example usage: _create_lookup_service(partitions_dir, ntypes, id_map, rank, world_size, num_parts) and returns: id_lookup