From 6558d3d2e00e05c89b6c74a8ba9652fce048d972 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 2 Sep 2020 17:58:04 +0200
Subject: [PATCH] [Test] skip non-existing end2end cases
---
tests/end2end/test_end2end.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tests/end2end/test_end2end.py b/tests/end2end/test_end2end.py
index cf9501ff5..328aaa163 100644
--- a/tests/end2end/test_end2end.py
+++ b/tests/end2end/test_end2end.py
@@ -212,6 +212,8 @@ def get_golden_io_pair(topology, wbits, abits):
@pytest.mark.parametrize("topology", ["tfc", "cnv"])
class TestEnd2End:
def test_export(self, topology, wbits, abits):
+ if wbits > abits:
+ pytest.skip("No wbits > abits end2end network configs for now")
(model, ishape) = get_trained_network_and_ishape(topology, wbits, abits)
chkpt_name = get_checkpoint_name(topology, wbits, abits, "export")
bo.export_finn_onnx(model, ishape, chkpt_name)
--
GitLab