From 6105bed454ea05a6111daaffcec64a68599a40fe Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 11 Sep 2020 13:01:15 +0200
Subject: [PATCH] [Test] fix test streamline fc for new FC nets
---
tests/transformation/streamline/test_streamline_fc.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/transformation/streamline/test_streamline_fc.py b/tests/transformation/streamline/test_streamline_fc.py
index dd7e756b4..9ce98066c 100644
--- a/tests/transformation/streamline/test_streamline_fc.py
+++ b/tests/transformation/streamline/test_streamline_fc.py
@@ -82,7 +82,7 @@ def test_streamline_fc(size, wbits, abits):
model = model.transform(RemoveUnusedTensors())
assert len(model.graph.initializer) == 11
assert len(model.graph.value_info) == 21
- assert len(model.graph.quantization_annotation) == 18
+ assert len(model.graph.quantization_annotation) == 20
produced_ctx = oxe.execute_onnx(model, input_dict, True)
produced = produced_ctx[model.graph.output[0].name]
assert np.isclose(expected, produced, atol=1e-3).all()
--
GitLab