commit d36c3bb8d919753f4ce0f1183bc04d1ae5df5c6b
author Cathal Corbett <cathal.corbett@arm.com> Fri Dec 23 14:13:35 2022 +0000
committer Cathal Corbett <cathal.corbett@arm.com> Fri Dec 23 17:41:30 2022 +0000
tree 44c8017d4e4c1d4ffe3eeebd5efa5f918f0e4738
parent 0bb096d9cd11bec1a890066064c8a28c4a4fd6ee

IVGCVSW-7369 Fix failing PyArmNN test

  * test_optimizer_options_fail test failing due to BF16 check
    removed from OptimizerOptions() in IVGCVSW-7214

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ibce5c0b0324f6d8761ffcb3e84532ca2656fcbc0

python/pyarmnn/test/test_modeloption.py

diff --git a/python/pyarmnn/test/test_modeloption.py b/python/pyarmnn/test/test_modeloption.py
index a47d2da..4773d90 100644
--- a/python/pyarmnn/test/test_modeloption.py
+++ b/python/pyarmnn/test/test_modeloption.py
@@ -118,17 +118,6 @@
 
     assert "Wrong number or type of arguments" in str(err.value)
 
-    with pytest.raises(RuntimeError) as err:
-        OptimizerOptions(True,
-                         False,
-                         True,
-                         ShapeInferenceMethod_InferAndValidate,
-                         True,
-                         [a],
-                         True)
-
-    assert "BFloat16 and Float16 optimization cannot be enabled at the same time" in str(err.value)
-
     with pytest.raises(TypeError) as err:
         oo = OptimizerOptions(True,
                               False,