Enabled ruff rule PT001 and ANN204

Author: Spaarsh

examples/python-udaf.py

@@ -26,7 +26,7 @@ class MyAccumulator(Accumulator):
     Interface of a user-defined accumulation.
     """
 
-    def __init__(self):
+    def __init__(self) -> None:
         self._sum = pyarrow.scalar(0.0)
 
     def update(self, values: pyarrow.Array) -> None:

examples/sql-using-python-udaf.py

@@ -25,7 +25,7 @@ class MyAccumulator(Accumulator):
     Interface of a user-defined accumulation.
     """
 
-    def __init__(self):
+    def __init__(self) -> None:
         self._sum = pa.scalar(0.0)
 
     def update(self, values: pa.Array) -> None:

pyproject.toml

@@ -80,8 +80,8 @@ ignore = [
     "TD003",   # Allow TODO lines
     "UP007",   # Disallowing Union is pedantic
     # TODO: Enable all of the following, but this PR is getting too large already
-    "PT001",
-    "ANN204",
+    #"PT001",
+    # "ANN204",
     "B008",
     "EM101",
     "PLR0913",

python/datafusion/expr.py

@@ -610,7 +610,7 @@ def over(self, window: Window) -> Expr:
 
 
 class ExprFuncBuilder:
-    def __init__(self, builder: expr_internal.ExprFuncBuilder):
+    def __init__(self, builder: expr_internal.ExprFuncBuilder) -> None:
         self.builder = builder
 
     def order_by(self, *exprs: Expr) -> ExprFuncBuilder: