From 5711b578af8428bf65f1e5c243ed10f5e3e8fb26 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 11 Sep 2020 23:29:53 +0200
Subject: [PATCH] [Util] do test checkpoint load/skip differently
---
src/finn/util/test.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/util/test.py b/src/finn/util/test.py
index e53d03736..3cd4248c5 100644
--- a/src/finn/util/test.py
+++ b/src/finn/util/test.py
@@ -88,10 +88,10 @@ def soft_verify_topk(invec, idxvec, k):
def load_test_checkpoint_or_skip(filename):
"Try to load given .onnx and return ModelWrapper, else skip current test."
- try:
+ if os.path.isfile(filename):
model = ModelWrapper(filename)
return model
- except FileNotFoundError:
+ else:
warnings.warn(filename + " not found from previous test step, skipping")
pytest.skip(filename + " not found from previous test step, skipping")
--
GitLab