|
91 | 91 | Extension = expr_internal.Extension |
92 | 92 | FileType = expr_internal.FileType |
93 | 93 | Filter = expr_internal.Filter |
94 | | -_GroupingSetInternal = expr_internal.GroupingSet |
95 | 94 | Join = expr_internal.Join |
96 | 95 | ILike = expr_internal.ILike |
97 | 96 | InList = expr_internal.InList |
@@ -1478,7 +1477,7 @@ def rollup(*exprs: Expr) -> Expr: |
1478 | 1477 | :py:func:`~datafusion.functions.grouping` |
1479 | 1478 | """ |
1480 | 1479 | args = [e.expr for e in exprs] |
1481 | | - return Expr(_GroupingSetInternal.rollup(*args)) |
| 1480 | + return Expr(expr_internal.GroupingSet.rollup(*args)) |
1482 | 1481 |
|
1483 | 1482 | @staticmethod |
1484 | 1483 | def cube(*exprs: Expr) -> Expr: |
@@ -1516,7 +1515,7 @@ def cube(*exprs: Expr) -> Expr: |
1516 | 1515 | :py:func:`~datafusion.functions.grouping` |
1517 | 1516 | """ |
1518 | 1517 | args = [e.expr for e in exprs] |
1519 | | - return Expr(_GroupingSetInternal.cube(*args)) |
| 1518 | + return Expr(expr_internal.GroupingSet.cube(*args)) |
1520 | 1519 |
|
1521 | 1520 | @staticmethod |
1522 | 1521 | def grouping_sets(*expr_lists: list[Expr]) -> Expr: |
@@ -1561,4 +1560,4 @@ def grouping_sets(*expr_lists: list[Expr]) -> Expr: |
1561 | 1560 | :py:func:`~datafusion.functions.grouping` |
1562 | 1561 | """ |
1563 | 1562 | 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