Fix typing retry 2

pandas-dev · Dec 12, 2024 · 042a4a9 · 042a4a9
1 parent 7d552ef
commit 042a4a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pandas/core/window/rolling.py b/pandas/core/window/rolling.py
@@ -652,9 +652,9 @@ def _numba_apply(
     def aggregate(self, func=None, *args, **kwargs):
         relabeling, func, columns, order = reconstruct_func(func, **kwargs)
         result = ResamplerWindowApply(self, func, args=args, kwargs=kwargs).agg()
-        if result is not None and isinstance(result, ABCDataFrame) and relabeling:
+        if (result is not None) and isinstance(result, ABCDataFrame) and relabeling:
             result = result.iloc[:, order]
-            result = result.set_axis(columns, axis=1)
+            result.columns = columns
         if result is None:
             return self.apply(func, raw=False, args=args, kwargs=kwargs)
         return result