Skip to content

Commit 6e76219

Browse files
fix usage of amended method
1 parent a7a0399 commit 6e76219

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

bigframes/core/array_value.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -319,11 +319,8 @@ def compute_general_reduction(
319319
is applied. Defaults to False.
320320
321321
Returns:
322-
Tuple[ArrayValue, Tuple[str, ...]]: A tuple containing:
323-
- An `ArrayValue` wrapping the new root node representing the
324-
aggregation/group-by result.
325-
- A tuple of strings representing the unique column IDs assigned to the
326-
resulting aggregate columns.
322+
ArrayValue:
323+
The new root node representing the aggregation/group-by result.
327324
"""
328325
plan = self.node
329326

@@ -357,8 +354,7 @@ def _is_direct_agg(agg_expr):
357354
new_root = expression_factoring.apply_agg_exprs_to_plan(
358355
plan, named_exprs, grouping_keys=[ex.deref(by) for by in by_column_ids]
359356
)
360-
target_ids = tuple(named_expr.id for named_expr in named_exprs)
361-
return (ArrayValue(new_root), target_ids)
357+
return ArrayValue(new_root)
362358

363359
def project_to_id(self, expression: ex.Expression):
364360
array_val, ids = self.compute_values(

0 commit comments

Comments
 (0)