From c03ba18c491e52cc85d8aae1825dd9e0b4f75e32 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Thu, 26 Oct 2023 19:21:54 +0200 Subject: Merging upstream version 18.17.0. Signed-off-by: Daniel Baumann --- sqlglot/dataframe/sql/functions.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'sqlglot/dataframe') 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: -- cgit v1.2.3