We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f7e558 commit 92e1108Copy full SHA for 92e1108
1 file changed
monai/apps/nnunet/nnunet_bundle.py
@@ -177,9 +177,9 @@ def __init__(
177
predictor: object,
178
model_folder: Union[str, Path],
179
model_name: str = "model.pt",
180
- dataset_json: dict = None,
181
- plans: dict = None,
182
- nnunet_config: dict = None,
+ dataset_json: Optional[dict] = None,
+ plans: Optional[dict] = None,
+ nnunet_config: Optional[dict] = None,
183
): # type: ignore
184
super().__init__()
185
self.predictor = predictor
@@ -331,9 +331,9 @@ def forward(self, x: MetaTensor) -> MetaTensor:
331
def get_nnunet_monai_predictor(
332
333
334
335
336
337
) -> ModelnnUNetWrapper:
338
"""
339
Initializes and returns a `nnUNetMONAIModelWrapper` containing the corresponding `nnUNetPredictor`.
0 commit comments