diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-02-16 05:45:52 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-02-16 05:45:52 +0000 |
commit | 3d48060515ba25b4c49d975a520ee0682327d1b7 (patch) | |
tree | e8730f509026e866d77c459f74a384505425363a /sqlglot/dataframe/sql | |
parent | Releasing debian version 21.0.2-1. (diff) | |
download | sqlglot-3d48060515ba25b4c49d975a520ee0682327d1b7.tar.xz sqlglot-3d48060515ba25b4c49d975a520ee0682327d1b7.zip |
Merging upstream version 21.1.1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'sqlglot/dataframe/sql')
-rw-r--r-- | sqlglot/dataframe/sql/functions.py | 2 | ||||
-rw-r--r-- | sqlglot/dataframe/sql/session.py | 10 |
2 files changed, 5 insertions, 7 deletions
diff --git a/sqlglot/dataframe/sql/functions.py b/sqlglot/dataframe/sql/functions.py index 29e7c55..133979a 100644 --- a/sqlglot/dataframe/sql/functions.py +++ b/sqlglot/dataframe/sql/functions.py @@ -148,7 +148,7 @@ def atanh(col: ColumnOrName) -> Column: def cbrt(col: ColumnOrName) -> Column: - return Column.invoke_anonymous_function(col, "CBRT") + return Column.invoke_expression_over_column(col, expression.Cbrt) def ceil(col: ColumnOrName) -> Column: diff --git a/sqlglot/dataframe/sql/session.py b/sqlglot/dataframe/sql/session.py index f518ac2..bfc022b 100644 --- a/sqlglot/dataframe/sql/session.py +++ b/sqlglot/dataframe/sql/session.py @@ -70,12 +70,10 @@ class SparkSession: column_mapping = {f"_{i}": None for i in range(1, len(data[0]) + 1)} data_expressions = [ - exp.Tuple( - expressions=list( - map( - lambda x: F.lit(x).expression, - row if not isinstance(row, dict) else row.values(), - ) + exp.tuple_( + *map( + lambda x: F.lit(x).expression, + row if not isinstance(row, dict) else row.values(), ) ) for row in data |