Skip to content

Commit 6d02dbc

Browse files
fix test_windowing
1 parent 75bb933 commit 6d02dbc

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

tests/system/small/engines/test_windowing.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,14 @@
1515
from google.cloud import bigquery
1616
import pytest
1717

18-
from bigframes.core import array_value, expression, identifiers, nodes, window_spec
18+
from bigframes.core import (
19+
array_value,
20+
expression,
21+
expression_types,
22+
identifiers,
23+
nodes,
24+
window_spec,
25+
)
1926
import bigframes.operations.aggregations as agg_ops
2027
from bigframes.session import direct_gbq_execution, polars_executor
2128
from bigframes.testing.engine_utils import assert_equivalence_execution
@@ -48,7 +55,9 @@ def test_engines_with_rows_window(
4855
)
4956
window_node = nodes.WindowOpNode(
5057
child=scalars_array_value.node,
51-
expression=expression.UnaryAggregation(agg_op, expression.deref("int64_too")),
58+
expression=expression_types.UnaryAggregation(
59+
agg_op, expression.deref("int64_too")
60+
),
5261
window_spec=window,
5362
output_name=identifiers.ColumnId("agg_int64"),
5463
never_skip_nulls=never_skip_nulls,

0 commit comments

Comments
 (0)