Skip to content

Commit 6222c5e

Browse files
committed
update rewrite logic
1 parent 87f265b commit 6222c5e

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

bigframes/core/rewrite/windows.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,13 @@
2020
from bigframes.core import nodes
2121

2222

23-
def rewrite_range_rolling(root: nodes.BigFrameNode) -> nodes.BigFrameNode:
24-
if isinstance(root, nodes.WindowOpNode):
25-
return _rewrite_range_rolling_node(root)
23+
def rewrite_range_rolling(node: nodes.BigFrameNode) -> nodes.BigFrameNode:
24+
if not isinstance(node, nodes.WindowOpNode):
25+
return node
2626

27-
return root
27+
if node.window_spec.row_bounded:
28+
return node
2829

29-
30-
def _rewrite_range_rolling_node(node: nodes.WindowOpNode) -> nodes.BigFrameNode:
3130
if len(node.window_spec.ordering) != 1:
3231
raise ValueError(
3332
"Range rolling should only be performed on exactly one column."

0 commit comments

Comments
 (0)