diff options
Diffstat (limited to 'python/pyarmnn/test/test_modeloption.py')
-rw-r--r-- | python/pyarmnn/test/test_modeloption.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/python/pyarmnn/test/test_modeloption.py b/python/pyarmnn/test/test_modeloption.py index a47d2da358..c03d4a8cce 100644 --- a/python/pyarmnn/test/test_modeloption.py +++ b/python/pyarmnn/test/test_modeloption.py @@ -71,8 +71,7 @@ def test_optimizer_options_with_model_opt(): False, ShapeInferenceMethod_InferAndValidate, True, - [a], - True) + [a]) mo = oo.m_ModelOptions @@ -113,8 +112,7 @@ def test_optimizer_options_fail(): False, ShapeInferenceMethod_InferAndValidate, True, - a, - True) + a) assert "Wrong number or type of arguments" in str(err.value) @@ -124,8 +122,7 @@ def test_optimizer_options_fail(): True, ShapeInferenceMethod_InferAndValidate, True, - [a], - True) + [a]) assert "BFloat16 and Float16 optimization cannot be enabled at the same time" in str(err.value) |