@@ -624,7 +624,7 @@ def __call__(self, *args: Expr) -> Expr:
624624 @overload
625625 @staticmethod
626626 def udwf (
627- input_type : pa .DataType | list [pa .DataType ],
627+ input_types : pa .DataType | list [pa .DataType ],
628628 return_type : pa .DataType ,
629629 volatility : Volatility | str ,
630630 name : Optional [str ] = None ,
@@ -634,7 +634,7 @@ def udwf(
634634 @staticmethod
635635 def udwf (
636636 func : Callable [[], WindowEvaluator ],
637- input_type : pa .DataType | list [pa .DataType ],
637+ input_types : pa .DataType | list [pa .DataType ],
638638 return_type : pa .DataType ,
639639 volatility : Volatility | str ,
640640 name : Optional [str ] = None ,
@@ -702,13 +702,13 @@ def biased_numbers() -> BiasedNumbers:
702702 def _create_window_udf (
703703 func : Callable [[], WindowEvaluator ],
704704 input_types : pa .DataType | list [pa .DataType ],
705- return_type : _R ,
705+ return_type : pa . DataType ,
706706 volatility : Volatility | str ,
707707 name : Optional [str ] = None ,
708708 ) -> WindowUDF :
709709 """Create a WindowUDF instance from function arguments."""
710710 if not callable (func ):
711- msg = "`func` argument must be callable"
711+ msg = "`func` must be callable. "
712712 raise TypeError (msg )
713713 if not isinstance (func (), WindowEvaluator ):
714714 msg = "`func` must implement the abstract base class WindowEvaluator"
0 commit comments