code: def get_n_rows(ddf, n):
      if n == -1:
          df = ddf.compute()
      else:
          df = ddf.head(n)
      return df
  
file_path: cugraph/benchmarks/shared/build_cugraph_ucx/test_client_bandwidth.py
imports: from dask_cuda import LocalCUDACluster
  from dask.distributed import Client, wait
  import cupy
  import numpy
  import cudf
  import dask_cudf
  import rmm
  from time import perf_counter_ns
label_desc: Use image node_id 3 for calling a global function with example usage: get_n_rows(ddf, n) and returns: df