Wrap Type to the typing of writer parameter of SaveImage

monai/transforms/io/array.py

@@ -20,7 +20,7 @@
 import warnings
 from pathlib import Path
 from pydoc import locate
-from typing import Dict, List, Optional, Sequence, Union
+from typing import Dict, List, Optional, Sequence, Type, Union
 
 import numpy as np
 import torch
@@ -330,7 +330,7 @@ def __init__(
         separate_folder: bool = True,
         print_log: bool = True,
         output_format: str = "",
-        writer: Union[image_writer.ImageWriter, str, None] = None,
+        writer: Union[Type[image_writer.ImageWriter], str, None] = None,
         channel_dim: Optional[int] = 0,
     ) -> None:
         self.folder_layout = FolderLayout(

monai/transforms/io/dictionary.py

@@ -16,7 +16,7 @@
 """
 
 from pathlib import Path
-from typing import Optional, Union
+from typing import Optional, Type, Union
 
 import numpy as np
 
@@ -237,7 +237,7 @@ def __init__(
         separate_folder: bool = True,
         print_log: bool = True,
         output_format: str = "",
-        writer: Union[image_writer.ImageWriter, str, None] = None,
+        writer: Union[Type[image_writer.ImageWriter], str, None] = None,
     ) -> None:
         super().__init__(keys, allow_missing_keys)
         self.meta_keys = ensure_tuple_rep(meta_keys, len(self.keys))

monai/transforms/utils_pytorch_numpy_unification.py

@@ -276,7 +276,7 @@ def cumsum(a: NdarrayOrTensor, axis=None, **kwargs) -> NdarrayOrTensor:
     """
 
     if isinstance(a, np.ndarray):
-        return np.cumsum(a, axis)
+        return np.cumsum(a, axis)  # type: ignore
     if axis is None:
         return torch.cumsum(a[:], 0, **kwargs)
     return torch.cumsum(a, dim=axis, **kwargs)