@@ -747,6 +747,7 @@ def wrapper(*args: Any, **kwargs: Any) -> Expr:
747747
748748 return decorator
749749
750+
750751class TableFunction :
751752 """Class for performing user-defined table functions (UDTF).
752753
@@ -755,9 +756,9 @@ class TableFunction:
755756 """
756757
757758 def __init__ (
758- self ,
759- name : str ,
760- func : Callable [[], any ],
759+ self ,
760+ name : str ,
761+ func : Callable [[], any ],
761762 ) -> None :
762763 """Instantiate a user-defined table function (UDTF).
763764
@@ -774,14 +775,14 @@ def __call__(self, *args: Expr) -> Any:
774775 @overload
775776 @staticmethod
776777 def udtf (
777- name : str ,
778+ name : str ,
778779 ) -> Callable [..., Any ]: ...
779780
780781 @overload
781782 @staticmethod
782783 def udtf (
783- func : Callable [[], Any ],
784- name : str ,
784+ func : Callable [[], Any ],
785+ name : str ,
785786 ) -> TableFunction : ...
786787
787788 @staticmethod
@@ -798,8 +799,8 @@ def udtf(*args: Any, **kwargs: Any):
798799
799800 @staticmethod
800801 def _create_table_udf (
801- func : Callable [..., Any ],
802- name : str ,
802+ func : Callable [..., Any ],
803+ name : str ,
803804 ) -> TableFunction :
804805 """Create a TableFunction instance from function arguments."""
805806 if not callable (func ):
@@ -810,7 +811,7 @@ def _create_table_udf(
810811
811812 @staticmethod
812813 def _create_table_udf_decorator (
813- name : Optional [str ] = None ,
814+ name : Optional [str ] = None ,
814815 ) -> Callable [[Callable [[], WindowEvaluator ]], Callable [..., Expr ]]:
815816 """Create a decorator for a WindowUDF."""
816817
0 commit comments