Skip to content

Commit 4169cd2

Browse files
committed
Fix NXP test
1 parent fe82a7a commit 4169cd2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/nxp/tests/test_batch_norm_fusion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def test_batch_norm_conv_fusing(bias: bool, input_shape: list[int]):
105105
og_nodes = list(program.graph.nodes)
106106
transformed_nodes = list(graph_module_out.graph.nodes)
107107

108-
assert len(og_nodes) == (11 if bias else 10)
108+
assert len(og_nodes) == (10 if bias else 9)
109109
assert og_nodes[9 if bias else 8].target.__name__ == "batch_norm.default"
110110

111111
assert len(transformed_nodes) == 5
@@ -139,7 +139,7 @@ def test_batch_norm_linear_fusing(bias: bool):
139139
og_nodes = list(og_module.graph.nodes)
140140
transformed_nodes = list(graph_module_out.graph.nodes)
141141

142-
assert len(og_nodes) == (11 if bias else 10)
142+
assert len(og_nodes) == (10 if bias else 9)
143143
assert og_nodes[8 if bias else 7].target.__name__ == "linear.default"
144144

145145
assert len(transformed_nodes) == 5

0 commit comments

Comments
 (0)