From b1fe0302d6b2d6ec4d32700a9e1b799a14ec032b Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Fri, 15 Nov 2019 18:30:47 +0000
Subject: [PATCH] [Test] get streamline transform name correctly
---
tests/test_streamline.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_streamline.py b/tests/test_streamline.py
index 75b030107..d0e83e8b5 100644
--- a/tests/test_streamline.py
+++ b/tests/test_streamline.py
@@ -68,7 +68,7 @@ def test_streamline_lfc_w1a1():
model = model.transform(InferDataTypes())
produced_ctx = oxe.execute_onnx(model, input_dict, True)
produced = produced_ctx[model.graph.output[0].name]
- # model.save("%d-%s.onnx" % (trn_ind, trn.__name__))
+ # model.save("%d-%s.onnx" % (trn_ind, trn.__class__.__name__))
assert np.isclose(expected, produced, atol=1e-3).all()
trn_ind += 1
os.remove(export_onnx_path)
--
GitLab