summaryrefslogtreecommitdiffstats
path: root/sqlglot/dataframe/sql/session.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlglot/dataframe/sql/session.py')
-rw-r--r--sqlglot/dataframe/sql/session.py20
1 files changed, 7 insertions, 13 deletions
diff --git a/sqlglot/dataframe/sql/session.py b/sqlglot/dataframe/sql/session.py
index af589b0..b883359 100644
--- a/sqlglot/dataframe/sql/session.py
+++ b/sqlglot/dataframe/sql/session.py
@@ -87,15 +87,13 @@ class SparkSession:
select_kwargs = {
"expressions": sel_columns,
"from": exp.From(
- expressions=[
- exp.Values(
- expressions=data_expressions,
- alias=exp.TableAlias(
- this=exp.to_identifier(self._auto_incrementing_name),
- columns=[exp.to_identifier(col_name) for col_name in column_mapping],
- ),
+ this=exp.Values(
+ expressions=data_expressions,
+ alias=exp.TableAlias(
+ this=exp.to_identifier(self._auto_incrementing_name),
+ columns=[exp.to_identifier(col_name) for col_name in column_mapping],
),
- ],
+ ),
),
}
@@ -128,10 +126,6 @@ class SparkSession:
return name
@property
- def _random_name(self) -> str:
- return "r" + uuid.uuid4().hex
-
- @property
def _random_branch_id(self) -> str:
id = self._random_id
self.known_branch_ids.add(id)
@@ -145,7 +139,7 @@ class SparkSession:
@property
def _random_id(self) -> str:
- id = self._random_name
+ id = "r" + uuid.uuid4().hex
self.known_ids.add(id)
return id