commit 626bd90378670eb5fd76f94526395430b752ad9e
author Francis Murtagh <francis.murtagh@arm.com> Tue Jun 21 13:16:23 2022 +0000
committer Francis Murtagh <francis.murtagh@arm.com> Wed Jun 22 08:36:28 2022 +0000
tree fa3c41a9ed7506f0bf5eb2b940f8661d2bf47adf
parent b9570c1a5b938aea20964e3b6a8164e9bf096b4c

Revert "Revert "IVGCVSW-6873 Import inputs but don't export outputs fails.""

This reverts commit a0f8b15d4ddb5075f380003ff31b271d389d3b66.

Reason for revert: <Test ClDmaBufInternalTests review >

Change-Id: Ibc4a77fa008643849da7330391942e4c87b941e2

python/pyarmnn/test/test_modeloption.py

diff --git a/python/pyarmnn/test/test_modeloption.py b/python/pyarmnn/test/test_modeloption.py
index c03d4a8..a47d2da 100644
--- a/python/pyarmnn/test/test_modeloption.py
+++ b/python/pyarmnn/test/test_modeloption.py
@@ -71,7 +71,8 @@
                           False,
                           ShapeInferenceMethod_InferAndValidate,
                           True,
-                          [a])
+                          [a],
+                          True)
 
     mo = oo.m_ModelOptions
 
@@ -112,7 +113,8 @@
                          False,
                          ShapeInferenceMethod_InferAndValidate,
                          True,
-                         a)
+                         a,
+                         True)
 
     assert "Wrong number or type of arguments" in str(err.value)
 
@@ -122,7 +124,8 @@
                          True,
                          ShapeInferenceMethod_InferAndValidate,
                          True,
-                         [a])
+                         [a],
+                         True)
 
     assert "BFloat16 and Float16 optimization cannot be enabled at the same time" in str(err.value)