code: def get_small_dataset(output_type):
      ary = [[1.0, 4.0, 4.0], [2.0, 2.0, 2.0], [5.0, 1.0, 1.0]]
      ary = cp.asarray(ary)
  
      if output_type == "numba":
          return as_cuda_array(ary)
  
      elif output_type == "cupy":
          return ary
  
      elif output_type == "numpy":
          return cp.asnumpy(ary)
  
      elif output_type == "pandas":
          return cudf.DataFrame(ary).to_pandas()
  
      else:
          return cudf.DataFrame(ary)
  
file_path: cuml/python/cuml/tests/test_module_config.py
imports: from numba.cuda import is_cuda_array, as_cuda_array
  from cuml.internals.safe_imports import cpu_only_import
  import cuml
  import pytest
  from cuml.internals.safe_imports import gpu_only_import
label_desc: Use image node_id 5 for calling a global function with example usage: get_small_dataset(output_type) and returns: as_cuda_array, ary, cp, cudf, cudf