Skip to content

Commit 48d53a4

Browse files
Fix documentation for output tensor return in ModelnnUNetWrapper
Signed-off-by: simben <[email protected]>
1 parent 2da5ca9 commit 48d53a4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/bundle/nnunet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,7 @@ def forward(self, x):
279279
- If the input is a collated batch, the filenames are extracted from the metadata of the input tensor.
280280
- The filenames are used to generate predictions using the nnUNet predictor.
281281
- The predictions are converted to torch tensors, with added batch and channel dimensions.
282-
- The output tensor is concatenated along the batch dimension and returned as a MetaTensor with the same metadata as the input.
282+
- The output tensor is concatenated along the batch dimension and returned as a MetaTensor with the same metadata.
283283
"""
284284
if type(x) is tuple: # if batch is decollated (list of tensors)
285285
input_files = [img.meta["filename_or_obj"][0] for img in x]

0 commit comments

Comments
 (0)