From 4f2b39613fc3ad6f2d1d54ef61a4e9cede32f1e7 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 3 Sep 2019 17:34:22 +0100
Subject: [PATCH] check for BIPOLAR qnt types in Brevitas export test
---
tests/test_brevitas_export.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/test_brevitas_export.py b/tests/test_brevitas_export.py
index d0df91f74..23d8af3d1 100644
--- a/tests/test_brevitas_export.py
+++ b/tests/test_brevitas_export.py
@@ -95,7 +95,7 @@ def test_brevitas_to_onnx_export():
int_weights_onnx = nph.to_array(model.graph.node[13].attribute[0].t)
assert (int_weights_onnx == int_weights_pytorch).all()
assert model.graph.node[12].attribute[0].name == "activation_qnt"
- assert model.graph.node[12].attribute[0].s.decode("utf-8") == "1"
+ assert model.graph.node[12].attribute[0].s.decode("utf-8") == "BIPOLAR"
assert model.graph.node[14].attribute[1].name == "weight_qnt"
- assert model.graph.node[14].attribute[1].s.decode("utf-8") == "1"
+ assert model.graph.node[14].attribute[1].s.decode("utf-8") == "BIPOLAR"
os.remove(export_onnx_path)
--
GitLab