code: def test_umap_mnmg(
      n_parts,
      n_rows,
      sampling_ratio,
      supervised,
      dataset,
      n_neighbors,
      client,
  ):
      result = _run_mnmg_test(
          n_parts,
          n_rows,
          sampling_ratio,
          supervised,
          dataset,
          n_neighbors,
          client,
      )
  
      if not result:
          result = _run_mnmg_test(
              n_parts,
              n_rows,
              sampling_ratio,
              supervised,
              dataset,
              n_neighbors,
              client,
          )
  
      assert result
  
file_path: cuml/python/cuml/tests/dask/test_dask_umap.py
imports: from sklearn.datasets import load_iris
  from sklearn.datasets import load_digits
  import math
  from cuml.metrics import trustworthiness
  from cuml.internals import logger
  from cuml.internals.safe_imports import cpu_only_import
  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: test_umap_mnmg(n_parts, n_rows, sampling_ratio, supervised, dataset, n_neighbors, client) without return types