fix: tighten scalar UDF return type hints (#1642)

This PR tightens the scalar UDF annotations in python/datafusion/user_defined.py:

the callable return type is bounded to pa.Array
the scalar UDF constructor accepts return_field: pa.Field
the decorator form accepts return_field: pa.DataType | pa.Field

Author: BharatDeva

python/datafusion/user_defined.py

@@ -249,7 +249,8 @@ def udf(*args: Any, **kwargs: Any):  # noqa: D417
             input_fields (list[pa.Field | pa.DataType]): The data types or Fields
                 of the arguments to ``func``. This list must be of the same length
                 as the number of arguments.
-            return_field (_R): The field of the return value from the function.
+            return_field (pa.DataType | pa.Field): The field of the return value
+                from the function.
             volatility (Volatility | str): See `Volatility` for allowed values.
             name (Optional[str]): A descriptive name for the function.