Skip to content

Commit 4d14faf

Browse files
author
Hoang Phan
committed
Add fix for using reducer with window with test added
1 parent 2d4bcb9 commit 4d14faf

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

quixstreams/dataframe/windows/aggregations.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -348,6 +348,7 @@ def __init__(
348348
reducer: Callable[[R, Any], R],
349349
initializer: Callable[[Any], R],
350350
) -> None:
351+
super().__init__()
351352
self._initializer: Callable[[Any], R] = initializer
352353
self._reducer: Callable[[R, Any], R] = reducer
353354

tests/test_quixstreams/test_dataframe/test_windows/test_aggregations.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,7 @@ def test_other_aggregation(self, aggregator, values, expected):
176176
(Mean(), "Mean"),
177177
(Max(), "Max"),
178178
(Min(), "Min"),
179+
(Reduce(reducer=lambda old, new: old + new, initializer=lambda x: x), "Reduce"),
179180
(Count("value"), "Count/value"),
180181
(Sum("value"), "Sum/value"),
181182
(Mean("value"), "Mean/value"),

0 commit comments

Comments
 (0)