Updated the _function and _decorator helpers in ScalarUDF.udf to use concrete Callable[..., Any] and pa.DataType | pa.Field annotations, removing the lingering references to the deleted _R type variable.

kosiew · kosiew · commit e0bce84be019 · 2025-10-22T18:34:54.000+08:00
diff --git a/python/datafusion/user_defined.py b/python/datafusion/user_defined.py
@@ -261,9 +261,9 @@ def double_udf(x):
         """
 
         def _function(
-            func: Callable[..., _R],
+            func: Callable[..., Any],
             input_types: list[pa.DataType],
-            return_type: _R,
+            return_type: pa.DataType | pa.Field,
             volatility: Volatility | str,
             name: Optional[str] = None,
         ) -> ScalarUDF:
@@ -285,7 +285,7 @@ def _function(
 
         def _decorator(
             input_types: list[pa.DataType],
-            return_type: _R,
+            return_type: pa.DataType | pa.Field,
             volatility: Volatility | str,
             name: Optional[str] = None,
         ) -> Callable:
