Skip to content

Commit 92e1108

Browse files
Update type hints for dataset_json, plans, and nnunet_config parameters to Optional[dict] in ModelnnUNetWrapper
1 parent 6f7e558 commit 92e1108

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

monai/apps/nnunet/nnunet_bundle.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -177,9 +177,9 @@ def __init__(
177177
predictor: object,
178178
model_folder: Union[str, Path],
179179
model_name: str = "model.pt",
180-
dataset_json: dict = None,
181-
plans: dict = None,
182-
nnunet_config: dict = None,
180+
dataset_json: Optional[dict] = None,
181+
plans: Optional[dict] = None,
182+
nnunet_config: Optional[dict] = None,
183183
): # type: ignore
184184
super().__init__()
185185
self.predictor = predictor
@@ -331,9 +331,9 @@ def forward(self, x: MetaTensor) -> MetaTensor:
331331
def get_nnunet_monai_predictor(
332332
model_folder: Union[str, Path],
333333
model_name: str = "model.pt",
334-
dataset_json: dict = None,
335-
plans: dict = None,
336-
nnunet_config: dict = None,
334+
dataset_json: Optional[dict] = None,
335+
plans: Optional[dict] = None,
336+
nnunet_config: Optional[dict] = None,
337337
) -> ModelnnUNetWrapper:
338338
"""
339339
Initializes and returns a `nnUNetMONAIModelWrapper` containing the corresponding `nnUNetPredictor`.

0 commit comments

Comments
 (0)