code: def get_test_data(size):
      return get_data_slice(size, size + 1)
  
file_path: scikit-learn-intelex/examples/sklearnex/kmeans_spmd.py
imports: from warnings import warn
  import dpctl.tensor
  import numpy
  from dpctl import SyclQueue
  from mpi4py import MPI
  from sklearn.datasets import load_digits
  from onedal.spmd.cluster import KMeans
label_desc: Use image node_id 3 for calling a global function with example usage: get_test_data(size) and returns: get_data_slice