diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py index 6d0e028a1dc499154e6526be9384f92438a4b98a..d0a9aa1c073d4f93e48b0d7d1ece17e1c288f56a 100644 --- a/tests/end2end/test_end2end_bnn_pynq.py +++ b/tests/end2end/test_end2end_bnn_pynq.py @@ -307,8 +307,8 @@ class TestEnd2End: def test_export(self, topology, wbits, abits): if wbits > abits: pytest.skip("No wbits > abits end2end network configs for now") - if topology == "lfc" and wbits > 1: - pytest.skip("Skipping non-existing lfc configs") + if topology == "lfc" and not (wbits == 1 and abits == 1): + pytest.skip("Skipping certain lfc configs") (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)