From b3fabb52f16f22faaa4664bbe3c6a0a12d064ebf Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Thu, 14 May 2020 18:50:07 +0100
Subject: [PATCH] [Test] fix test cond in infer_datatypes
---
tests/transformation/test_infer_datatypes.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/tests/transformation/test_infer_datatypes.py b/tests/transformation/test_infer_datatypes.py
index 77b6a94f8..e3db40289 100644
--- a/tests/transformation/test_infer_datatypes.py
+++ b/tests/transformation/test_infer_datatypes.py
@@ -54,8 +54,8 @@ def test_infer_datatypes():
assert model.get_tensor_datatype("MatMul_1_out0") == DataType.INT32
assert model.get_tensor_datatype("MatMul_2_out0") == DataType.INT32
assert model.get_tensor_datatype("MatMul_3_out0") == DataType.INT32
- assert model.get_tensor_datatype("Sign_0_out0") == DataType.BIPOLAR
- assert model.get_tensor_datatype("Sign_1_out0") == DataType.BIPOLAR
- assert model.get_tensor_datatype("Sign_2_out0") == DataType.BIPOLAR
- assert model.get_tensor_datatype("Sign_3_out0") == DataType.BIPOLAR
+ assert model.get_tensor_datatype("MultiThreshold_0_out0") == DataType.BIPOLAR
+ assert model.get_tensor_datatype("MultiThreshold_1_out0") == DataType.BIPOLAR
+ assert model.get_tensor_datatype("MultiThreshold_2_out0") == DataType.BIPOLAR
+ assert model.get_tensor_datatype("MultiThreshold_3_out0") == DataType.BIPOLAR
os.remove(export_onnx_path)
--
GitLab