From 744631a28d0c2daece2a63c65ca861c26a7bd0cb Mon Sep 17 00:00:00 2001 From: Robert Kalmar Date: Thu, 17 Jul 2025 11:31:54 +0200 Subject: [PATCH] NXP backend: Fix unittest-nxp-neutron introduced in 0012ffaf9 --- .../ir/converter/node_converter/test_max_pool_2d_converter.py | 3 ++- backends/nxp/tests/test_node_format_inference.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/backends/nxp/tests/ir/converter/node_converter/test_max_pool_2d_converter.py b/backends/nxp/tests/ir/converter/node_converter/test_max_pool_2d_converter.py index 2618558f7c9..50bbf100980 100644 --- a/backends/nxp/tests/ir/converter/node_converter/test_max_pool_2d_converter.py +++ b/backends/nxp/tests/ir/converter/node_converter/test_max_pool_2d_converter.py @@ -61,7 +61,8 @@ def test_max_pool_2d_conversion(input_shape, padding): # Otherwise, we get violation that this op is not part of ATen Core ops. edge_program._verifiers = [ EXIREdgeDialectVerifier( - class_only=True, exception_list=[torch.ops.aten.max_pool2d.default] + class_only=True, + core_aten_ops_exception_list=[torch.ops.aten.max_pool2d.default], ) ] diff --git a/backends/nxp/tests/test_node_format_inference.py b/backends/nxp/tests/test_node_format_inference.py index 96107efa755..e2796187ce8 100644 --- a/backends/nxp/tests/test_node_format_inference.py +++ b/backends/nxp/tests/test_node_format_inference.py @@ -71,7 +71,8 @@ def test_maxpool2d(): # Otherwise, we get violation that this op is not part of ATen Core ops. edge_program._verifiers = [ EXIREdgeDialectVerifier( - class_only=True, exception_list=[torch.ops.aten.max_pool2d.default] + class_only=True, + core_aten_ops_exception_list=[torch.ops.aten.max_pool2d.default], ) ]