Updated the internal ScalarUDF.udf helper signatures so IDEs and type checkers surface support for pyarrow.Field inputs when defining UDFs

Author: kosiew

python/datafusion/user_defined.py

@@ -262,7 +262,7 @@ def double_udf(x):
 
         def _function(
             func: Callable[..., Any],
-            input_types: list[pa.DataType],
+            input_types: list[pa.DataType | pa.Field],
             return_type: pa.DataType | pa.Field,
             volatility: Volatility | str,
             name: Optional[str] = None,
@@ -284,7 +284,7 @@ def _function(
             )
 
         def _decorator(
-            input_types: list[pa.DataType],
+            input_types: list[pa.DataType | pa.Field],
             return_type: pa.DataType | pa.Field,
             volatility: Volatility | str,
             name: Optional[str] = None,