From 3fb3589b149757cf827dcac488ff113be2d64fcc Mon Sep 17 00:00:00 2001 From: Matthew Murray Date: Thu, 26 Sep 2024 05:46:19 -0700 Subject: [PATCH] use private API _column_names --- python/cudf/cudf/core/window/rolling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cudf/cudf/core/window/rolling.py b/python/cudf/cudf/core/window/rolling.py index 043a41145e5..e30938ab892 100644 --- a/python/cudf/cudf/core/window/rolling.py +++ b/python/cudf/cudf/core/window/rolling.py @@ -24,7 +24,7 @@ class _RollingBase: def _apply_agg_dataframe(self, df, agg_name): result_df = cudf.DataFrame({}) - for i, col_name in enumerate(df.columns): + for i, col_name in enumerate(df._column_names): result_col = self._apply_agg_series(df[col_name], agg_name) result_df.insert(i, col_name, result_col) result_df.index = df.index