Skip to content

Commit

Permalink
Resolve error on registering udaf #874
Browse files Browse the repository at this point in the history
  • Loading branch information
timsaucer committed Sep 20, 2024
1 parent 47b2c8b commit 2ba0b9a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/datafusion/udf.py
Original file line number Diff line number Diff line change
Expand Up @@ -192,7 +192,7 @@ def __init__(
See :py:func:`udaf` for a convenience function and argument
descriptions.
"""
self._udf = df_internal.AggregateUDF(
self._udaf = df_internal.AggregateUDF(
name, accumulator, input_types, return_type, state_type, str(volatility)
)

Expand All @@ -203,7 +203,7 @@ def __call__(self, *args: Expr) -> Expr:
occur during the evaluation of the dataframe.
"""
args = [arg.expr for arg in args]
return Expr(self._udf.__call__(*args))
return Expr(self._udaf.__call__(*args))

@staticmethod
def udaf(
Expand Down

0 comments on commit 2ba0b9a

Please sign in to comment.