Skip to content

Commit 32b97a0

Browse files
committed
fix mypy
1 parent 9a56bb5 commit 32b97a0

File tree

2 files changed

+11
-14
lines changed

2 files changed

+11
-14
lines changed

bigframes/core/blocks.py

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1169,20 +1169,18 @@ def explode(
11691169
) -> Block:
11701170
expr = self.expr.explode(column_ids)
11711171
if ignore_index:
1172-
new_index_col_id = guid.generate_guid("explode_index_")
1173-
expr = expr.promote_offsets(new_index_col_id)
1174-
expr = expr.drop_columns(self.index_columns)
1175-
index_columns = [new_index_col_id]
1176-
index_labels = [None]
1172+
return Block(
1173+
expr.drop_columns(self.index_columns),
1174+
column_labels=self.column_labels,
1175+
index_columns=[],
1176+
)
11771177
else:
1178-
index_columns = list(self.index_columns)
1179-
index_labels = self.column_labels.names
1180-
return Block(
1181-
expr,
1182-
column_labels=self.column_labels,
1183-
index_columns=index_columns,
1184-
index_labels=index_labels,
1185-
)
1178+
return Block(
1179+
expr,
1180+
column_labels=self.column_labels,
1181+
index_columns=self.index_columns,
1182+
index_labels=self.column_labels.names,
1183+
)
11861184

11871185
def _standard_stats(self, column_id) -> typing.Sequence[agg_ops.UnaryAggregateOp]:
11881186
"""

third_party/bigframes_vendored/ibis/expr/operations/generic.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,5 @@
55
from ibis.expr.operations.core import Unary
66

77

8-
# TODO: add this function to ibis
98
class GenerateArray(Unary):
109
dtype = dt.Array(dt.int64)

0 commit comments

Comments
 (0)