Arm backend: Fix test_lstm_arm.py after mypy changes

perheld · perheld · commit 63ed767fac74 · 2025-11-04T15:58:11.000+01:00
Use keywords when calling pipeline.change_args()

Signed-off-by: per.held@arm.com
Change-Id: Iba70b4c26eb1b96c16ee47382002cd8f7c36852b
diff --git 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()
 

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,9 @@ def test_lstm_tosa_FP():`
`51`	`51`	`exir_op=[],`
`52`	`52`	`use_to_edge_transform_and_lower=True,`
`53`	`53`	`)`
`54`		`- pipeline.change_args("run_method_and_compare_outputs", get_test_inputs(), atol=3e-1)`
	`54`	`+ pipeline.change_args(`
	`55`	`+ "run_method_and_compare_outputs", inputs=get_test_inputs(), atol=3e-1`
	`56`	`+ )`
`55`	`57`	`pipeline.run()`
`56`	`58`
`57`	`59`
`@@ -64,7 +66,10 @@ def test_lstm_tosa_INT():`
`64`	`66`	`use_to_edge_transform_and_lower=True,`
`65`	`67`	`)`
`66`	`68`	`pipeline.change_args(`
`67`		`- "run_method_and_compare_outputs", get_test_inputs(), atol=3e-1, qtol=1.0`
	`69`	`+ "run_method_and_compare_outputs",`
	`70`	`+ inputs=get_test_inputs(),`
	`71`	`+ atol=3e-1,`
	`72`	`+ qtol=1.0,`
`68`	`73`	`)`
`69`	`74`	`pipeline.run()`
`70`	`75`
`@@ -79,7 +84,10 @@ def test_lstm_u55_INT():`
`79`	`84`	`use_to_edge_transform_and_lower=True,`
`80`	`85`	`)`
`81`	`86`	`pipeline.change_args(`
`82`		`- "run_method_and_compare_outputs", get_test_inputs(), atol=3e-1, qtol=1.0`
	`87`	`+ "run_method_and_compare_outputs",`
	`88`	`+ inputs=get_test_inputs(),`
	`89`	`+ atol=3e-1,`
	`90`	`+ qtol=1.0,`
`83`	`91`	`)`
`84`	`92`	`pipeline.run()`
`85`	`93`
`@@ -94,7 +102,10 @@ def test_lstm_u85_INT():`
`94`	`102`	`use_to_edge_transform_and_lower=True,`
`95`	`103`	`)`
`96`	`104`	`pipeline.change_args(`
`97`		`- "run_method_and_compare_outputs", get_test_inputs(), atol=3e-1, qtol=1.0`
	`105`	`+ "run_method_and_compare_outputs",`
	`106`	`+ inputs=get_test_inputs(),`
	`107`	`+ atol=3e-1,`
	`108`	`+ qtol=1.0,`
`98`	`109`	`)`
`99`	`110`	`pipeline.run()`
`100`	`111`