code: def _clean_directory(path):
      """param <path> could either be relative or absolute."""
      if os.path.isfile(path):
          os.remove(path)  # remove the file
      elif os.path.isdir(path):
          shutil.rmtree(path)
  
file_path: cugraph/python/cugraph-dgl/cugraph_dgl/dataloading/dataloader.py
imports: from __future__ import annotations
  import os
  import shutil
  import cugraph_dgl
  import cupy
  import cudf
  from cugraph.utilities.utils import import_optional
  from cugraph.gnn import BulkSampler
  from dask.distributed import default_client, Event
  from cugraph_dgl.dataloading import HomogenousBulkSamplerDataset, HeterogenousBulkSamplerDataset
  from cugraph_dgl.dataloading.utils.extract_graph_helpers import create_cugraph_graph_from_edges_dict
label_desc: Use image node_id 4 for calling a global function with example usage: _clean_directory(path) without return types