def METHOD_NAME(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