diff options
Diffstat (limited to 'sqlglot/dataframe')
-rw-r--r-- | sqlglot/dataframe/sql/functions.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/sqlglot/dataframe/sql/functions.py b/sqlglot/dataframe/sql/functions.py index d98feee..a424ea4 100644 --- a/sqlglot/dataframe/sql/functions.py +++ b/sqlglot/dataframe/sql/functions.py @@ -84,11 +84,11 @@ def min(col: ColumnOrName) -> Column: def max_by(col: ColumnOrName, ord: ColumnOrName) -> Column: - return Column.invoke_anonymous_function(col, "MAX_BY", ord) + return Column.invoke_expression_over_column(col, expression.ArgMax, expression=ord) def min_by(col: ColumnOrName, ord: ColumnOrName) -> Column: - return Column.invoke_anonymous_function(col, "MIN_BY", ord) + return Column.invoke_expression_over_column(col, expression.ArgMin, expression=ord) def count(col: ColumnOrName) -> Column: @@ -1113,7 +1113,7 @@ def reverse(col: ColumnOrName) -> Column: def flatten(col: ColumnOrName) -> Column: - return Column.invoke_anonymous_function(col, "FLATTEN") + return Column.invoke_expression_over_column(col, expression.Flatten) def map_keys(col: ColumnOrName) -> Column: |