code: def ucx_client(ucx_cluster):
      client = Client(ucx_cluster)
      yield client
      client.close()
  
file_path: cuml/python/cuml/tests/dask/conftest.py
imports: import pytest
  from dask_cuda import initialize
  from dask_cuda import LocalCUDACluster
  from dask_cuda.utils_test import IncreasedCloseTimeoutNanny
  from dask.distributed import Client
label_desc: Use image node_id 4 for calling a global function with example usage: ucx_client(ucx_cluster) without return types