use private API _column_names

rapidsai · Sep 26, 2024 · 3fb3589 · 3fb3589
1 parent 7885580
commit 3fb3589
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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