code: def ucx_cluster():
      initialize.initialize(
          create_cuda_context=True,
          enable_tcp_over_ucx=enable_tcp_over_ucx,
          enable_nvlink=enable_nvlink,
          enable_infiniband=enable_infiniband,
      )
      cluster = LocalCUDACluster(
          protocol="ucx",
          enable_tcp_over_ucx=enable_tcp_over_ucx,
          enable_nvlink=enable_nvlink,
          enable_infiniband=enable_infiniband,
          worker_class=IncreasedCloseTimeoutNanny,
      )
      yield cluster
      cluster.close()
  
file_path: cuml/python/cuml/benchmark/automated/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 3 for calling a global function with example usage: ucx_cluster() without return types