@@ -800,7 +800,7 @@ class ReadLocalNode(LeafNode):
800
800
# Mapping of local ids to bfet id.
801
801
scan_list : ScanList
802
802
# Offsets are generated only if this is non-null
803
- offsets_col : Optional [bigframes . core . identifiers .ColumnId ] = None
803
+ offsets_col : Optional [identifiers .ColumnId ] = None
804
804
session : typing .Optional [bigframes .session .Session ] = None
805
805
806
806
@property
@@ -1039,7 +1039,7 @@ class CachedTableNode(ReadTableNode):
1039
1039
# Unary nodes
1040
1040
@dataclasses .dataclass (frozen = True , eq = False )
1041
1041
class PromoteOffsetsNode (UnaryNode , AdditiveNode ):
1042
- col_id : bigframes . core . identifiers .ColumnId
1042
+ col_id : identifiers .ColumnId
1043
1043
1044
1044
@property
1045
1045
def non_local (self ) -> bool :
@@ -1317,9 +1317,7 @@ def remap_refs(
1317
1317
class ProjectionNode (UnaryNode , AdditiveNode ):
1318
1318
"""Assigns new variables (without modifying existing ones)"""
1319
1319
1320
- assignments : typing .Tuple [
1321
- typing .Tuple [ex .Expression , bigframes .core .identifiers .ColumnId ], ...
1322
- ]
1320
+ assignments : typing .Tuple [typing .Tuple [ex .Expression , identifiers .ColumnId ], ...]
1323
1321
1324
1322
def _validate (self ):
1325
1323
input_types = self .child ._dtype_lookup
@@ -1445,9 +1443,7 @@ def remap_refs(
1445
1443
1446
1444
@dataclasses .dataclass (frozen = True , eq = False )
1447
1445
class AggregateNode (UnaryNode ):
1448
- aggregations : typing .Tuple [
1449
- typing .Tuple [ex .Aggregation , bigframes .core .identifiers .ColumnId ], ...
1450
- ]
1446
+ aggregations : typing .Tuple [typing .Tuple [ex .Aggregation , identifiers .ColumnId ], ...]
1451
1447
by_column_ids : typing .Tuple [ex .DerefOp , ...] = tuple ([])
1452
1448
order_by : Tuple [OrderingExpression , ...] = ()
1453
1449
dropna : bool = True
@@ -1539,7 +1535,7 @@ def remap_refs(
1539
1535
class WindowOpNode (UnaryNode , AdditiveNode ):
1540
1536
expression : ex .Aggregation
1541
1537
window_spec : window .WindowSpec
1542
- output_name : bigframes . core . identifiers .ColumnId
1538
+ output_name : identifiers .ColumnId
1543
1539
never_skip_nulls : bool = False
1544
1540
skip_reproject_unsafe : bool = False
1545
1541
@@ -1682,7 +1678,7 @@ def remap_refs(
1682
1678
class ExplodeNode (UnaryNode ):
1683
1679
column_ids : typing .Tuple [ex .DerefOp , ...]
1684
1680
# Offsets are generated only if this is non-null
1685
- offsets_col : Optional [bigframes . core . identifiers .ColumnId ] = None
1681
+ offsets_col : Optional [identifiers .ColumnId ] = None
1686
1682
1687
1683
@property
1688
1684
def row_preserving (self ) -> bool :
0 commit comments