Fix of model 0007 error in debug logging mode.

RichardObi · Nov 28, 2023 · fa286d1 · fa286d1
1 parent 093ba76
commit fa286d1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 16 deletions.
diff --git 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
@@ -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},