This is a breaking change to move the module datafusion.udf to datafusion.user_defined

Author: timsaucer

docs/source/user-guide/common-operations/udf-and-udfa.rst

@@ -26,7 +26,7 @@ Scalar Functions
 
 When writing a user-defined function that can operate on a row by row basis, these are called Scalar
 Functions. You can define your own scalar function by calling
-:py:func:`~datafusion.udf.ScalarUDF.udf` .
+:py:func:`~datafusion.user_defined.ScalarUDF.udf` .
 
 The basic definition of a scalar UDF is a python function that takes one or more
 `pyarrow <https://arrow.apache.org/docs/python/index.html>`_ arrays and returns a single array as
@@ -93,9 +93,9 @@ converting to Python objects to do the evaluation.
 Aggregate Functions
 -------------------
 
-The :py:func:`~datafusion.udf.AggregateUDF.udaf` function allows you to define User-Defined
+The :py:func:`~datafusion.user_defined.AggregateUDF.udaf` function allows you to define User-Defined
 Aggregate Functions (UDAFs). To use this you must implement an
-:py:class:`~datafusion.udf.Accumulator` that determines how the aggregation is performed.
+:py:class:`~datafusion.user_defined.Accumulator` that determines how the aggregation is performed.
 
 When defining a UDAF there are four methods you need to implement. The ``update`` function takes the
 array(s) of input and updates the internal state of the accumulator. You should define this function
@@ -153,8 +153,8 @@ Window Functions
 ----------------
 
 To implement a User-Defined Window Function (UDWF) you must call the
-:py:func:`~datafusion.udf.WindowUDF.udwf` function using a class that implements the abstract
-class :py:class:`~datafusion.udf.WindowEvaluator`.
+:py:func:`~datafusion.user_defined.WindowUDF.udwf` function using a class that implements the abstract
+class :py:class:`~datafusion.user_defined.WindowEvaluator`.
 
 There are three methods of evaluation of UDWFs.
 
@@ -207,7 +207,7 @@ determine which evaluate functions are called.
 
     import pyarrow as pa
     from datafusion import udwf, col, SessionContext
-    from datafusion.udf import WindowEvaluator
+    from datafusion.user_defined import WindowEvaluator
 
     class ExponentialSmooth(WindowEvaluator):
         def __init__(self, alpha: float) -> None:

examples/python-udwf.py

@@ -22,7 +22,7 @@
 from datafusion import col, lit, udwf
 from datafusion import functions as f
 from datafusion.expr import WindowFrame
-from datafusion.udf import WindowEvaluator
+from datafusion.user_defined import WindowEvaluator
 
 # This example creates five different examples of user defined window functions in order
 # to demonstrate the variety of ways a user may need to implement.

python/datafusion/__init__.py

@@ -49,7 +49,15 @@
 from .io import read_avro, read_csv, read_json, read_parquet
 from .plan import ExecutionPlan, LogicalPlan
 from .record_batch import RecordBatch, RecordBatchStream
-from .udf import Accumulator, AggregateUDF, ScalarUDF, WindowUDF, udaf, udf, udwf
+from .user_defined import (
+    Accumulator,
+    AggregateUDF,
+    ScalarUDF,
+    WindowUDF,
+    udaf,
+    udf,
+    udwf,
+)
 
 __version__ = importlib_metadata.version(__name__)
 

python/datafusion/context.py

@@ -30,7 +30,7 @@
 from datafusion.dataframe import DataFrame
 from datafusion.expr import Expr, SortExpr, sort_list_to_raw_sort_list
 from datafusion.record_batch import RecordBatchStream
-from datafusion.udf import AggregateUDF, ScalarUDF, WindowUDF
+from datafusion.user_defined import AggregateUDF, ScalarUDF, WindowUDF
 
 from ._internal import RuntimeEnvBuilder as RuntimeEnvBuilderInternal
 from ._internal import SessionConfig as SessionConfigInternal

python/datafusion/udf.py python/datafusion/user_defined.pypython/datafusion/udf.py renamed to python/datafusion/user_defined.py

@@ -107,7 +107,7 @@ def test_class_module_is_datafusion():
         AggregateUDF,
         ScalarUDF,
     ]:
-        assert klass.__module__ == "datafusion.udf"
+        assert klass.__module__ == "datafusion.user_defined"
 
     # expressions
     for klass in [Expr, Column, Literal, BinaryExpr, AggregateFunction]:

python/tests/test_imports.py

@@ -22,7 +22,7 @@
 from datafusion import SessionContext, column, lit, udwf
 from datafusion import functions as f
 from datafusion.expr import WindowFrame
-from datafusion.udf import WindowEvaluator
+from datafusion.user_defined import WindowEvaluator
 
 
 class ExponentialSmoothDefault(WindowEvaluator):