From c4821927ed826aa6b2447d981cf4ff5c7b1c9b9e Mon Sep 17 00:00:00 2001 From: Per Held Date: Tue, 4 Nov 2025 15:17:40 +0100 Subject: [PATCH] Arm backend: Fix test_lstm_arm.py after mypy changes Use keywords when calling pipeline.change_args() Signed-off-by: per.held@arm.com Change-Id: Iba70b4c26eb1b96c16ee47382002cd8f7c36852b --- backends/arm/test/models/test_lstm_arm.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/backends/arm/test/models/test_lstm_arm.py b/backends/arm/test/models/test_lstm_arm.py index 6ee16b6a31a..d9691efab25 100644 --- a/backends/arm/test/models/test_lstm_arm.py +++ b/backends/arm/test/models/test_lstm_arm.py @@ -51,7 +51,9 @@ def test_lstm_tosa_FP(): exir_op=[], use_to_edge_transform_and_lower=True, ) - pipeline.change_args("run_method_and_compare_outputs", get_test_inputs(), atol=3e-1) + pipeline.change_args( + "run_method_and_compare_outputs", inputs=get_test_inputs(), atol=3e-1 + ) pipeline.run() @@ -64,7 +66,10 @@ def test_lstm_tosa_INT(): use_to_edge_transform_and_lower=True, ) pipeline.change_args( - "run_method_and_compare_outputs", get_test_inputs(), atol=3e-1, qtol=1.0 + "run_method_and_compare_outputs", + inputs=get_test_inputs(), + atol=3e-1, + qtol=1.0, ) pipeline.run() @@ -79,7 +84,10 @@ def test_lstm_u55_INT(): use_to_edge_transform_and_lower=True, ) pipeline.change_args( - "run_method_and_compare_outputs", get_test_inputs(), atol=3e-1, qtol=1.0 + "run_method_and_compare_outputs", + inputs=get_test_inputs(), + atol=3e-1, + qtol=1.0, ) pipeline.run() @@ -94,7 +102,10 @@ def test_lstm_u85_INT(): use_to_edge_transform_and_lower=True, ) pipeline.change_args( - "run_method_and_compare_outputs", get_test_inputs(), atol=3e-1, qtol=1.0 + "run_method_and_compare_outputs", + inputs=get_test_inputs(), + atol=3e-1, + qtol=1.0, ) pipeline.run()