Skip to content

Commit a7cad28

Browse files
Update nnUNet import paths and comment out Transposed transform in integration tests
1 parent 2e6bb14 commit a7cad28

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integration/test_integration_nnunet_bundle.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
from monai.apps.nnunet import nnUNetV2Runner
2222
from monai.bundle.config_parser import ConfigParser
23-
from monai.bundle.nnunet import convert_nnunet_to_monai_bundle, get_nnunet_monai_predictor, get_nnunet_trainer
23+
from monai.apps.nnunet.nnunet_bundle import convert_nnunet_to_monai_bundle, get_nnunet_monai_predictor, get_nnunet_trainer
2424
from monai.data import DataLoader, Dataset, create_test_image_3d
2525
from monai.transforms import Compose, Decollated, EnsureChannelFirstd, LoadImaged, SaveImaged, Transposed
2626
from monai.utils import optional_import
@@ -129,7 +129,7 @@ def test_nnunet_bundle(self) -> None:
129129
post_processing_transforms = Compose(
130130
[
131131
Decollated(keys=None, detach=True),
132-
Transposed(keys="pred", indices=[0, 3, 2, 1]),
132+
#Transposed(keys="pred", indices=[0, 3, 2, 1]),
133133
SaveImaged(
134134
keys="pred", output_dir=Path(self.sim_dataroot).joinpath("predictions"), output_postfix="pred"
135135
),

0 commit comments

Comments
 (0)