code: def get_x_y(n_samples, n_classes):
      X = cudf.DataFrame({"x": range(n_samples)})
      y = cp.arange(n_samples) % n_classes
      cp.random.shuffle(y)
      y = cudf.Series(y)
      return X, y
  
imports: from cuml.model_selection import StratifiedKFold
  import pytest
  from cuml.internals.safe_imports import gpu_only_import
label_desc: Use image node_id 1 for calling a global function with example usage: get_x_y(n_samples, n_classes) and returns: X,  y