Skip to content

Commit 7dd08bd

Browse files
authored
Make InMemImageReader work when monai is not available (#169)
Signed-off-by: Gigon Bae <gbae@nvidia.com>
1 parent 6a0d60c commit 7dd08bd

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

monai/deploy/operators/monai_seg_inference_operator.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,11 @@
2020
np_str_obj_array_pattern, _ = optional_import("torch.utils.data._utils.collate", name="np_str_obj_array_pattern")
2121
Dataset, _ = optional_import("monai.data", name="Dataset")
2222
DataLoader, _ = optional_import("monai.data", name="DataLoader")
23-
ImageReader_, _ = optional_import("monai.data", name="ImageReader")
23+
ImageReader_, image_reader_ok_ = optional_import("monai.data", name="ImageReader")
2424
# Dynamic class is not handled so make it Any for now: https://github.com/python/mypy/issues/2477
2525
ImageReader: Any = ImageReader_
26+
if not image_reader_ok_:
27+
ImageReader = object # for 'class InMemImageReader(ImageReader):' to work
2628
decollate_batch, _ = optional_import("monai.data", name="decollate_batch")
2729
sliding_window_inference, _ = optional_import("monai.inferers", name="sliding_window_inference")
2830
ensure_tuple, _ = optional_import("monai.utils", name="ensure_tuple")

0 commit comments

Comments
 (0)