fix rewriter bug

TrevorBergeron · TrevorBergeron · commit 30d43b805331 · 2025-12-12T03:27:14.000Z
diff --git a/bigframes/core/rewrite/sampling.py b/bigframes/core/rewrite/sampling.py
@@ -97,7 +97,7 @@ def hash_based_sample(
     uniq_row_id = ops.RowHash().as_expr(
         ex.DerefOp(rowhash_col_id), ex.DerefOp(dupe_count_col_id), ex.const(seed)
     )
-    result = nodes.ProjectionNode(root, ((uniq_row_id, unique_row_id_col_id),))
+    result = nodes.ProjectionNode(result, ((uniq_row_id, unique_row_id_col_id),))
 
     if fraction < 1:
         # The filtering is correlated with the ordering, but thats fine because the ordering is pseudo-random

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def hash_based_sample(`
`97`	`97`	`uniq_row_id = ops.RowHash().as_expr(`
`98`	`98`	`ex.DerefOp(rowhash_col_id), ex.DerefOp(dupe_count_col_id), ex.const(seed)`
`99`	`99`	`)`
`100`		`- result = nodes.ProjectionNode(root, ((uniq_row_id, unique_row_id_col_id),))`
	`100`	`+ result = nodes.ProjectionNode(result, ((uniq_row_id, unique_row_id_col_id),))`
`101`	`101`
`102`	`102`	`if fraction < 1:`
`103`	`103`	`# The filtering is correlated with the ordering, but thats fine because the ordering is pseudo-random`