Skip to content

Commit b7d360a

Browse files
fix identifier tests
1 parent 5717cfc commit b7d360a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

bigframes/core/rewrite/identifiers.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,5 +109,9 @@ def remap_variables(
109109
# have to do top down to preserve node identities
110110
return (
111111
root.top_down(_create_mapping_operator(id_def_remaps, id_ref_remaps)),
112-
id_def_remaps[root],
112+
# Only used by unit tests
113+
{
114+
old_id: (id_def_remaps[root] | id_ref_remaps[root])[old_id]
115+
for old_id in root.ids
116+
},
113117
)

0 commit comments

Comments
 (0)