refactor: Update udwf calls in WindowUDF to use BiasedNumbers directly

kosiew · kosiew · commit ae623835f07e · 2025-03-13T17:37:47.000+08:00
- Changed udwf1 to use BiasedNumbers instead of bias_10.
- Added udwf2 to call udwf with bias_10.
- Introduced udwf3 to demonstrate a lambda function returning BiasedNumbers(20).
diff --git a/python/datafusion/udf.py b/python/datafusion/udf.py
@@ -653,7 +653,10 @@ def evaluate_all(self, values: list[pa.Array], num_rows: int) -> pa.Array:
             def bias_10() -> BiasedNumbers:
                 return BiasedNumbers(10)
 
-            udwf1 = udwf(bias_10, pa.int64(), pa.int64(), "immutable")
+            udwf1 = udwf(BiasedNumbers, pa.int64(), pa.int64(), "immutable")
+            udwf2 = udwf(bias_10, pa.int64(), pa.int64(), "immutable")
+            udwf3 = udwf(lambda: BiasedNumbers(20), pa.int64(), pa.int64(), "immutable")
+
             ```
 
         **Decorator example:**
