diff --git a/examples/python-udwf.py b/examples/python-udwf.py index 05b3021d..55de2bdc 100644 --- a/examples/python-udwf.py +++ b/examples/python-udwf.py @@ -185,35 +185,36 @@ def evaluate_all(self, values: list[pa.Array], num_rows: int) -> pa.Array: df = ctx.create_dataframe([[batch]]) exp_smooth = udwf( - ExponentialSmoothDefault(0.9), + lambda: ExponentialSmoothDefault(0.9), pa.float64(), pa.float64(), volatility="immutable", ) smooth_two_row = udwf( - SmoothBoundedFromPreviousRow(0.9), + lambda: SmoothBoundedFromPreviousRow(0.9), pa.float64(), pa.float64(), volatility="immutable", ) smooth_rank = udwf( - SmoothAcrossRank(0.9), + lambda: SmoothAcrossRank(0.9), pa.float64(), pa.float64(), volatility="immutable", ) smooth_frame = udwf( - ExponentialSmoothFrame(0.9), + lambda: ExponentialSmoothFrame(0.9), pa.float64(), pa.float64(), volatility="immutable", + name="smooth_frame", ) smooth_two_col = udwf( - SmoothTwoColumn(0.9), + lambda: SmoothTwoColumn(0.9), [pa.float64(), pa.int64()], pa.float64(), volatility="immutable",