Skip to content

Commit 36fdd3f

Browse files
Fix formatting in docstring and remove unused import statements in MONetBundleInferenceOperator
Signed-off-by: Simone Bendazzoli <simben@kth.se>
1 parent 24a6500 commit 36fdd3f

File tree

2 files changed

+2
-21
lines changed

2 files changed

+2
-21
lines changed

examples/apps/ai_spleen_nnunet_seg_app/app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ class AISpleennnUNetSegApp(Application):
3434
"""Demonstrates inference with built-in MONet Bundle inference operator with DICOM files as input/output
3535
3636
This application loads a set of DICOM instances, select the appropriate series, converts the series to
37-
3D volume image, performs inference with the built-in MONet Bundle inference operator, including nnUNet resampling,pre-processing
37+
3D volume image, performs inference with the built-in MONet Bundle inference operator, including nnUNet resampling, pre-processing
3838
and post-processing, save the segmentation image in a DICOM Seg OID in an instance file, and optionally the
3939
surface mesh in STL format.
4040

monai/deploy/operators/monet_bundle_inference_operator.py

Lines changed: 1 addition & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -15,27 +15,8 @@
1515
from monai.deploy.operators.monai_bundle_inference_operator import MonaiBundleInferenceOperator, get_bundle_config
1616
from monai.deploy.utils.importutil import optional_import
1717

18-
MONAI_UTILS = "monai.utils"
19-
nibabel, _ = optional_import("nibabel", "3.2.1")
2018
torch, _ = optional_import("torch", "1.10.2")
2119

22-
NdarrayOrTensor, _ = optional_import("monai.config", name="NdarrayOrTensor")
23-
MetaTensor, _ = optional_import("monai.data.meta_tensor", name="MetaTensor")
24-
PostFix, _ = optional_import("monai.utils.enums", name="PostFix") # For the default meta_key_postfix
25-
first, _ = optional_import("monai.utils.misc", name="first")
26-
ensure_tuple, _ = optional_import(MONAI_UTILS, name="ensure_tuple")
27-
convert_to_dst_type, _ = optional_import(MONAI_UTILS, name="convert_to_dst_type")
28-
Key, _ = optional_import(MONAI_UTILS, name="ImageMetaKey")
29-
MetaKeys, _ = optional_import(MONAI_UTILS, name="MetaKeys")
30-
SpaceKeys, _ = optional_import(MONAI_UTILS, name="SpaceKeys")
31-
Compose_, _ = optional_import("monai.transforms", name="Compose")
32-
ConfigParser_, _ = optional_import("monai.bundle", name="ConfigParser")
33-
MapTransform_, _ = optional_import("monai.transforms", name="MapTransform")
34-
SimpleInferer, _ = optional_import("monai.inferers", name="SimpleInferer")
35-
36-
Compose: Any = Compose_
37-
MapTransform: Any = MapTransform_
38-
ConfigParser: Any = ConfigParser_
3920
__all__ = ["MONetBundleInferenceOperator"]
4021

4122

@@ -82,7 +63,7 @@ def predict(self, data: Any, *args, **kwargs) -> Union[Image, Any, Tuple[Any, ..
8263
"""Predicts output using the inferer."""
8364

8465
self._nnunet_predictor.predictor.network = self._model_network
85-
# os.environ['nnUNet_def_n_proc'] = "1"
66+
8667
if len(data.shape) == 4:
8768
data = data[None]
8869
return self._nnunet_predictor(data)

0 commit comments

Comments
 (0)