Skip to content

Commit c52b49c

Browse files
timsaucerclaude
andcommitted
Remove _GroupingSetInternal alias, use expr_internal.GroupingSet directly
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 1243498 commit c52b49c

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

python/datafusion/expr.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,6 @@
9191
Extension = expr_internal.Extension
9292
FileType = expr_internal.FileType
9393
Filter = expr_internal.Filter
94-
_GroupingSetInternal = expr_internal.GroupingSet
9594
Join = expr_internal.Join
9695
ILike = expr_internal.ILike
9796
InList = expr_internal.InList
@@ -1478,7 +1477,7 @@ def rollup(*exprs: Expr) -> Expr:
14781477
:py:func:`~datafusion.functions.grouping`
14791478
"""
14801479
args = [e.expr for e in exprs]
1481-
return Expr(_GroupingSetInternal.rollup(*args))
1480+
return Expr(expr_internal.GroupingSet.rollup(*args))
14821481

14831482
@staticmethod
14841483
def cube(*exprs: Expr) -> Expr:
@@ -1516,7 +1515,7 @@ def cube(*exprs: Expr) -> Expr:
15161515
:py:func:`~datafusion.functions.grouping`
15171516
"""
15181517
args = [e.expr for e in exprs]
1519-
return Expr(_GroupingSetInternal.cube(*args))
1518+
return Expr(expr_internal.GroupingSet.cube(*args))
15201519

15211520
@staticmethod
15221521
def grouping_sets(*expr_lists: list[Expr]) -> Expr:
@@ -1561,4 +1560,4 @@ def grouping_sets(*expr_lists: list[Expr]) -> Expr:
15611560
:py:func:`~datafusion.functions.grouping`
15621561
"""
15631562
raw_lists = [[e.expr for e in lst] for lst in expr_lists]
1564-
return Expr(_GroupingSetInternal.grouping_sets(*raw_lists))
1563+
return Expr(expr_internal.GroupingSet.grouping_sets(*raw_lists))

0 commit comments

Comments
 (0)