More ruff formatting suggestions

timsaucer · timsaucer · commit f67a72767bb6 · 2025-03-10T07:42:27.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -67,7 +67,13 @@ features = ["substrait"]
 # Enable docstring linting using the google style guide
 [tool.ruff.lint]
 select = ["ALL" ]
-ignore = ["COM812", "ISC001", "TD002"]  # Recommended to ignore these rules when using with ruff-format
+ignore = [
+    "ANN401",  # Allow Any for wrapper classes
+    "COM812",  # Recommended to ignore these rules when using with ruff-format
+    "ISC001",  # Recommended to ignore these rules when using with ruff-format
+    "TD002",
+    "UP007"    # Disallowing Union is pedantic
+]
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
diff --git a/python/datafusion/__init__.py b/python/datafusion/__init__.py
@@ -86,6 +86,9 @@
     "read_json",
     "read_parquet",
     "substrait",
+    "udaf",
+    "udf",
+    "udwf",
 ]
 
 
diff --git a/python/datafusion/udf.py b/python/datafusion/udf.py
