commit d5914a4f97fe26ea32a45b3f04b369d595518810
author Teresa Charlin <teresa.charlinreyes@arm.com> Thu Apr 20 12:38:29 2023 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Apr 20 12:46:49 2023 +0000
tree eff32355a99f7020f107a160d80fd07cbe949a45
parent 4cf40d70f8e9da3184bb4c9b604db6a3b23d6533

IVGCVSW-7685 PyArmNN test failures in nightly

* This was originated when setting scale to 1 in quantization parameters

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic3192a6985f3afc595952eb5bfdaa9b628bbb88d

python/pyarmnn/test/test_onnx_parser.py

diff --git a/python/pyarmnn/test/test_onnx_parser.py b/python/pyarmnn/test/test_onnx_parser.py
index 1424c24..b6e46f2 100644
--- a/python/pyarmnn/test/test_onnx_parser.py
+++ b/python/pyarmnn/test/test_onnx_parser.py
@@ -1,4 +1,4 @@
-# Copyright © 2020 Arm Ltd. All rights reserved.
+# Copyright © 2020,2023 Arm Ltd. All rights reserved.
 # SPDX-License-Identifier: MIT
 import os
 
@@ -45,7 +45,7 @@
     assert tensor.GetNumDimensions() == 4
     assert tensor.GetNumElements() == 784
     assert tensor.GetQuantizationOffset() == 0
-    assert tensor.GetQuantizationScale() == 0
+    assert tensor.GetQuantizationScale() == 1
 
 
 def test_onnx_parser_get_network_output_binding_info(parser):
@@ -56,7 +56,7 @@
     assert tensor.GetNumDimensions() == 4
     assert tensor.GetNumElements() == 10
     assert tensor.GetQuantizationOffset() == 0
-    assert tensor.GetQuantizationScale() == 0
+    assert tensor.GetQuantizationScale() == 1
 
 
 def test_onnx_filenotfound_exception(shared_data_folder):