code: def assert_membership_vectors(cu_vecs, sk_vecs):
      """
      Assert the membership vectors by taking the adjusted rand score
      of the argsorted membership vectors.
      """
      if sk_vecs.shape == cu_vecs.shape:
          cu_labels_sorted = np.argsort(cu_vecs)[::-1]
          sk_labels_sorted = np.argsort(sk_vecs)[::-1]
  
          k = min(sk_vecs.shape[1], 10)
          for i in range(k):
              assert (
                  adjusted_rand_score(
                      cu_labels_sorted[:, i], sk_labels_sorted[:, i]
                  )
                  >= 0.85
              )
  
imports: import platform
  from cuml.testing.test_preproc_utils import to_output_type
  from cuml.testing.utils import array_equal
  from cuml.cluster.hdbscan import HDBSCAN
  from cuml.neighbors import NearestNeighbors
  from cuml.metrics import trustworthiness
  from cuml.metrics import adjusted_rand_score
  from cuml.manifold import UMAP
  from cuml.linear_model import ElasticNet, Lasso, LinearRegression, LogisticRegression, Ridge
  from cuml.internals.memory_utils import using_memory_type
  from cuml.internals.mem_type import MemoryType
  from cuml.decomposition import PCA, TruncatedSVD
  from cuml.common.device_selection import DeviceType, using_device_type
  from hdbscan import HDBSCAN
  from sklearn.neighbors import NearestNeighbors
  from sklearn.linear_model import Ridge
  from sklearn.linear_model import ElasticNet
  from sklearn.linear_model import Lasso
  from sklearn.linear_model import LogisticRegression
  from sklearn.linear_model import LinearRegression
  from sklearn.decomposition import PCA
  from sklearn.decomposition import TruncatedSVD
  from sklearn.datasets import make_regression, make_blobs
  from pytest_cases import fixture_union, fixture_plus
  from importlib import import_module
  import inspect
  import pickle
  from cuml.internals.safe_imports import gpu_only_import
  import itertools
  import pytest
  import cuml
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 1 for calling a global function with example usage: assert_membership_vectors(cu_vecs, sk_vecs) without return types