code: def kpss_ref(y):
      """Wrapper around statsmodels' KPSS test"""
      batch_size = y.shape[1]
      test_results = np.zeros(batch_size, dtype=bool)
      for i in range(batch_size):
          with warnings.catch_warnings():
              warnings.filterwarnings("ignore")
              _, pval, *_ = stattools.kpss(
                  y[:, i], regression="c", nlags="legacy"
              )
          test_results[i] = pval > 0.05
      return test_results
  
imports: from cuml.tsa import stationarity
  from statsmodels.tsa import stattools
  import warnings
  import pytest
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 2 for calling a global function with example usage: kpss_ref(y) and returns: test_results