Skip to content

Commit 54e5e0d

Browse files
authored
fix: udwf example (#948)
1 parent 92b093c commit 54e5e0d

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

examples/python-udwf.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -185,35 +185,36 @@ def evaluate_all(self, values: list[pa.Array], num_rows: int) -> pa.Array:
185185
df = ctx.create_dataframe([[batch]])
186186

187187
exp_smooth = udwf(
188-
ExponentialSmoothDefault(0.9),
188+
lambda: ExponentialSmoothDefault(0.9),
189189
pa.float64(),
190190
pa.float64(),
191191
volatility="immutable",
192192
)
193193

194194
smooth_two_row = udwf(
195-
SmoothBoundedFromPreviousRow(0.9),
195+
lambda: SmoothBoundedFromPreviousRow(0.9),
196196
pa.float64(),
197197
pa.float64(),
198198
volatility="immutable",
199199
)
200200

201201
smooth_rank = udwf(
202-
SmoothAcrossRank(0.9),
202+
lambda: SmoothAcrossRank(0.9),
203203
pa.float64(),
204204
pa.float64(),
205205
volatility="immutable",
206206
)
207207

208208
smooth_frame = udwf(
209-
ExponentialSmoothFrame(0.9),
209+
lambda: ExponentialSmoothFrame(0.9),
210210
pa.float64(),
211211
pa.float64(),
212212
volatility="immutable",
213+
name="smooth_frame",
213214
)
214215

215216
smooth_two_col = udwf(
216-
SmoothTwoColumn(0.9),
217+
lambda: SmoothTwoColumn(0.9),
217218
[pa.float64(), pa.int64()],
218219
pa.float64(),
219220
volatility="immutable",

0 commit comments

Comments
 (0)