code: def prepare_worker_to_parts(data, client=None):
      if client is None:
          client = get_client()
      for placeholder, worker in enumerate(client.has_what().keys()):
          if worker not in data.worker_to_parts:
              data.worker_to_parts[worker] = [placeholder]
      return data
  
file_path: cugraph/python/cugraph/cugraph/dask/common/mg_utils.py
imports: import os
  import gc
  import numba.cuda
  from cugraph.dask.common.read_utils import MissingUCXPy
label_desc: Use image node_id 2 for calling a global function with example usage: prepare_worker_to_parts(data, client) and returns: data