From edc4dddae3d044f6f6e5358b6e3a6e082e98bd3b Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 20 May 2020 01:33:50 +0100
Subject: [PATCH] [Test] use canonical layout in MW test
---
tests/core/test_modelwrapper.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/core/test_modelwrapper.py b/tests/core/test_modelwrapper.py
index c04deb616..d1da6934a 100644
--- a/tests/core/test_modelwrapper.py
+++ b/tests/core/test_modelwrapper.py
@@ -31,7 +31,7 @@ import onnx
from collections import Counter
import brevitas.onnx as bo
import numpy as np
-import finn.core.data_layout as data_layout
+import finn.core.data_layout as DataLayout
from finn.core.modelwrapper import ModelWrapper
from finn.util.test import get_test_model_trained
@@ -70,7 +70,7 @@ def test_modelwrapper():
assert out_prod.op_type == "MultiThreshold"
inp_layout = model.get_tensor_layout(inp_name)
assert inp_layout is None
- inp_layout = data_layout.NCHW
+ inp_layout = DataLayout.NCHW
model.set_tensor_layout(inp_name, inp_layout)
assert model.get_tensor_layout(inp_name) == inp_layout
os.remove(export_onnx_path)
--
GitLab