From 311be3961cba780feff4164aadb75e8a5d4a84a5 Mon Sep 17 00:00:00 2001 From: ambroslins Date: Mon, 24 May 2021 00:46:12 +0200 Subject: [PATCH] fix type for margin argument in CropForegroundd Signed-off-by: ambroslins --- monai/transforms/croppad/dictionary.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/transforms/croppad/dictionary.py b/monai/transforms/croppad/dictionary.py index faa96b1e4c..4ec14e0a7d 100644 --- a/monai/transforms/croppad/dictionary.py +++ b/monai/transforms/croppad/dictionary.py @@ -693,7 +693,7 @@ def __init__( source_key: str, select_fn: Callable = is_positive, channel_indices: Optional[IndexSelection] = None, - margin: int = 0, + margin: Union[Sequence[int], int] = 0, k_divisible: Union[Sequence[int], int] = 1, mode: Union[NumpyPadMode, str] = NumpyPadMode.CONSTANT, start_coord_key: str = "foreground_start_coord",