Fix ruff errors

Author: kosiew

python/datafusion/udf.py

@@ -19,7 +19,6 @@
 
 from __future__ import annotations
 
-from ast import Call
 import functools
 from abc import ABCMeta, abstractmethod
 from enum import Enum
@@ -647,15 +646,17 @@ def udwf(
     ) -> WindowUDF: ...
 
     @staticmethod
-    def udwf(*args: Any, **kwargs: Any):  # noqa: D417
+    def udwf(*args: Any, **kwargs: Any):  # noqa: D417, C901
         """Create a new User-Defined Window Function (UDWF).
 
         This class can be used both as a **function** and as a **decorator**.
 
         Usage:
-            - **As a function**: Call `udwf(func, input_types, return_type, volatility, name)`.
-            - **As a decorator**: Use `@udwf(input_types, return_type, volatility, name)`.
-                When using `udwf` as a decorator, **do not pass `func` explicitly**.
+            - **As a function**: Call `udwf(func, input_types, return_type, volatility,
+              name)`.
+            - **As a decorator**: Use `@udwf(input_types, return_type, volatility,
+              name)`. When using `udwf` as a decorator, **do not pass `func`
+              explicitly**.
 
         **Function example:**
             ```
@@ -665,7 +666,8 @@ class BiasedNumbers(WindowEvaluator):
                 def __init__(self, start: int = 0) -> None:
                     self.start = start
 
-                def evaluate_all(self, values: list[pa.Array], num_rows: int) -> pa.Array:
+                def evaluate_all(self, values: list[pa.Array],
+                    num_rows: int) -> pa.Array:
                     return pa.array([self.start + i for i in range(num_rows)])
 
             def bias_10() -> BiasedNumbers:
@@ -685,8 +687,8 @@ def biased_numbers() -> BiasedNumbers:
             ```
 
         Args:
-            func: **Only needed when calling as a function. Skip this argument when using
-                `udwf` as a decorator.**
+            func: **Only needed when calling as a function. Skip this argument when
+                using `udwf` as a decorator.**
             input_types: The data types of the arguments.
             return_type: The data type of the return value.
             volatility: See :py:class:`Volatility` for allowed values.

python/tests/test_udwf.py

@@ -303,7 +303,11 @@ def test_register_udwf(ctx, count_window_df):
 
     ctx.register_udwf(window_count)
     result = ctx.sql(
-        "SELECT window_count(a) OVER (ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING) FROM test_table"
+        """
+        SELECT window_count(a)
+        OVER (ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED
+        FOLLOWING) FROM test_table
+        """
     ).collect()[0]
     assert result.column(0) == pa.array([0, 1, 2])
 
@@ -487,6 +491,9 @@ def test_udwf_named_function(ctx, count_window_df):
 
     ctx.register_udwf(window_count)
     result = ctx.sql(
-        "SELECT my_custom_counter(a) OVER (ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING) FROM test_table"
+        """
+        SELECT my_custom_counter(a)
+        OVER (ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED
+        FOLLOWING) FROM test_table"""
     ).collect()[0]
     assert result.column(0) == pa.array([0, 1, 2])