refactor: Rename input_type to input_types in udwf method signature for clarity

kosiew · kosiew · commit 20d5dd9ff55b · 2025-03-13T18:55:13.000+08:00
diff --git a/python/datafusion/udf.py b/python/datafusion/udf.py
@@ -624,7 +624,7 @@ def __call__(self, *args: Expr) -> Expr:
     @overload
     @staticmethod
     def udwf(
-        input_type: pa.DataType | list[pa.DataType],
+        input_types: pa.DataType | list[pa.DataType],
         return_type: pa.DataType,
         volatility: Volatility | str,
         name: Optional[str] = None,
@@ -634,7 +634,7 @@ def udwf(
     @staticmethod
     def udwf(
         func: Callable[[], WindowEvaluator],
-        input_type: pa.DataType | list[pa.DataType],
+        input_types: pa.DataType | list[pa.DataType],
         return_type: pa.DataType,
         volatility: Volatility | str,
         name: Optional[str] = None,
@@ -702,13 +702,13 @@ def biased_numbers() -> BiasedNumbers:
     def _create_window_udf(
         func: Callable[[], WindowEvaluator],
         input_types: pa.DataType | list[pa.DataType],
-        return_type: _R,
+        return_type: pa.DataType,
         volatility: Volatility | str,
         name: Optional[str] = None,
     ) -> WindowUDF:
         """Create a WindowUDF instance from function arguments."""
         if not callable(func):
-            msg = "`func` argument must be callable"
+            msg = "`func` must be callable."
             raise TypeError(msg)
         if not isinstance(func(), WindowEvaluator):
             msg = "`func` must implement the abstract base class WindowEvaluator"
