From fa286d1f2cb2349748ff6967db5946e537d32ea1 Mon Sep 17 00:00:00 2001 From: RichardObi Date: Tue, 28 Nov 2023 20:43:43 +0100 Subject: [PATCH] Fix of model 0007 error in debug logging mode. --- config/global.json | 2 +- tests/test_model_executor.py | 30 +++++++++++++++--------------- 2 files changed, 16 insertions(+), 16 deletions(-) diff --git a/config/global.json b/config/global.json index 15bcfee..ac58884 100644 --- a/config/global.json +++ b/config/global.json @@ -952,7 +952,7 @@ "00007_INPAINT_BRAIN_MRI": { "execution": { "package_name": "00007_INPAINT_BRAIN_MRI", - "package_link": "https://zenodo.org/record/7042217/files/00007_INPAINT_BRAIN_MRI.zip?download=1", + "package_link": "https://zenodo.org/records/10214677/files/00007_INPAINT_BRAIN_MRI.zip?download=1", "model_name": "inp_gen", "extension": ".pth", "image_size": [ diff --git a/tests/test_model_executor.py b/tests/test_model_executor.py index 0aef62a..0873fcd 100644 --- a/tests/test_model_executor.py +++ b/tests/test_model_executor.py @@ -55,21 +55,21 @@ ), ("00005_DCGAN_MMG_MASS_ROI", {}, 3), ("00006_WGANGP_MMG_MASS_ROI", {}, 3), - # ( - # "00007_INPAINT_BRAIN_MRI", - # { - # "image_size": (256, 256), - # "num_inpaints_per_sample": 2, - # "randomize_input_image_order": False, - # "add_variations_to_mask": False, - # "x_center": 120, - # "y_center": 140, - # "radius_1": 8, - # "radius_2": 12, - # "radius_3": 24, - # }, - # 3, - # ), + ( + "00007_INPAINT_BRAIN_MRI", + { + "image_size": (256, 256), + "num_inpaints_per_sample": 2, + "randomize_input_image_order": False, + "add_variations_to_mask": False, + "x_center": 120, + "y_center": 140, + "radius_1": 8, + "radius_2": 12, + "radius_3": 24, + }, + 3, + ), ( "00008_C-DCGAN_MMG_MASSES", {"condition": 0, "is_cbisddsm_training_data": False},