From 1e90a57aeba564f1ff0ff18fe35521b33eea639c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= Date: Tue, 31 Oct 2023 15:08:27 +0100 Subject: [PATCH] Minor refactoring --- demo/src/gui.py | 1 - demo/src/inference.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/src/gui.py b/demo/src/gui.py index 7c6ec13..74df10d 100644 --- a/demo/src/gui.py +++ b/demo/src/gui.py @@ -11,7 +11,6 @@ from .utils import load_ct_to_numpy from .utils import load_pred_volume_to_numpy - # setup logging LOGGER = setup_logger() diff --git a/demo/src/inference.py b/demo/src/inference.py index 15a6ab1..5d3774c 100644 --- a/demo/src/inference.py +++ b/demo/src/inference.py @@ -91,6 +91,7 @@ def run_model( shutil.rmtree(patient_directory) if os.path.exists(output_path): shutil.rmtree(output_path) + except Exception: print(traceback.format_exc()) # Clean-up