From eebb284d3130e122f933f5241509d0a438699d9d Mon Sep 17 00:00:00 2001 From: Gigon Bae Date: Sat, 9 Oct 2021 11:06:18 -0700 Subject: [PATCH] Make InMemImageReader work when monai is not available Signed-off-by: Gigon Bae --- monai/deploy/operators/monai_seg_inference_operator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/monai/deploy/operators/monai_seg_inference_operator.py b/monai/deploy/operators/monai_seg_inference_operator.py index ec9d5c7c..cb028008 100644 --- a/monai/deploy/operators/monai_seg_inference_operator.py +++ b/monai/deploy/operators/monai_seg_inference_operator.py @@ -20,9 +20,11 @@ np_str_obj_array_pattern, _ = optional_import("torch.utils.data._utils.collate", name="np_str_obj_array_pattern") Dataset, _ = optional_import("monai.data", name="Dataset") DataLoader, _ = optional_import("monai.data", name="DataLoader") -ImageReader_, _ = optional_import("monai.data", name="ImageReader") +ImageReader_, image_reader_ok_ = optional_import("monai.data", name="ImageReader") # Dynamic class is not handled so make it Any for now: https://github.com/python/mypy/issues/2477 ImageReader: Any = ImageReader_ +if not image_reader_ok_: + ImageReader = object # for 'class InMemImageReader(ImageReader):' to work decollate_batch, _ = optional_import("monai.data", name="decollate_batch") sliding_window_inference, _ = optional_import("monai.inferers", name="sliding_window_inference") ensure_tuple, _ = optional_import("monai.utils", name="ensure_tuple")