code: def get_cuda_cluster():
      from dask_cuda import LocalCUDACluster
  
      CUDA_VISIBLE_DEVICES = os.environ.get("CUDA_VISIBLE_DEVICES", "0")
      n_workers = min(2, len(CUDA_VISIBLE_DEVICES.split(",")))
      cluster = LocalCUDACluster(n_workers=n_workers)
      yield cluster
      cluster.close()
  
file_path: Merlin/tests/conftest.py
imports: from __future__ import absolute_import
  import contextlib
  import os
  from pathlib import Path
  import pytest
  import platform
  import warnings
  from pathlib import Path
  import distributed
  import psutil
  import pytest
  from asvdb import BenchmarkInfo, utils
  from merlin.models.utils import ci_utils
label_desc: Use image node_id 2 for calling a global function with example usage: get_cuda_cluster() without return types