Merge branch 'dev' of https://github.com/Project-MONAI/model-zoo into…

… add-unittests
Project-MONAI · Aug 25, 2023 · 6ca465c · 6ca465c
2 parents 3095fe6 + 99b8be7
commit 6ca465c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/ci/bundle_custom_data.py b/ci/bundle_custom_data.py
@@ -62,5 +62,9 @@
     "endoscopic_inbody_classification": {},
     "spleen_deepedit_annotation": {},
     "spleen_ct_segmentation": {},
-    "lung_nodule_ct_detection": {},
+    "lung_nodule_ct_detection": {
+        "input_shape": [1, 1, 512, 512, 192],
+        "onnx_output_names": ["output_0", "output_1", "output_2", "output_3", "output_4", "output_5"],
+        "network_def#use_list_output": True,
+    },
 }
diff --git a/ci/utils.py b/ci/utils.py
@@ -18,10 +18,11 @@
 import subprocess
 from typing import List
 
-from github import Github
 from monai.apps.utils import download_url
 from monai.bundle.config_parser import ConfigParser
-from monai.utils import look_up_option
+from monai.utils import look_up_option, optional_import
+
+Github, _ = optional_import("github", name="Github")
 
 SUPPORTED_HASH_TYPES = {"md5": hashlib.md5, "sha1": hashlib.sha1, "sha256": hashlib.sha256, "sha512": hashlib.sha512}