From 626bd90378670eb5fd76f94526395430b752ad9e Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Tue, 21 Jun 2022 13:16:23 +0000 Subject: Revert "Revert "IVGCVSW-6873 Import inputs but don't export outputs fails."" This reverts commit a0f8b15d4ddb5075f380003ff31b271d389d3b66. Reason for revert: Change-Id: Ibc4a77fa008643849da7330391942e4c87b941e2 --- python/pyarmnn/test/test_modeloption.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'python/pyarmnn/test/test_modeloption.py') diff --git a/python/pyarmnn/test/test_modeloption.py b/python/pyarmnn/test/test_modeloption.py index c03d4a8cce..a47d2da358 100644 --- a/python/pyarmnn/test/test_modeloption.py +++ b/python/pyarmnn/test/test_modeloption.py @@ -71,7 +71,8 @@ def test_optimizer_options_with_model_opt(): False, ShapeInferenceMethod_InferAndValidate, True, - [a]) + [a], + True) mo = oo.m_ModelOptions @@ -112,7 +113,8 @@ def test_optimizer_options_fail(): False, ShapeInferenceMethod_InferAndValidate, True, - a) + a, + True) assert "Wrong number or type of arguments" in str(err.value) @@ -122,7 +124,8 @@ def test_optimizer_options_fail(): True, ShapeInferenceMethod_InferAndValidate, True, - [a]) + [a], + True) assert "BFloat16 and Float16 optimization cannot be enabled at the same time" in str(err.value) -- cgit v1.2.1