Skip to content

Commit b61e4e1

Browse files
Refactor nnUNet bundle integration tests for clarity and remove redundant method
1 parent ccdc76c commit b61e4e1

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

tests/test_integration_nnunet_bundle.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def setUp(self) -> None:
8686
self.test_path = test_path
8787

8888
@skip_if_no_cuda
89-
def test_nnunetBundle_get_trainer(self) -> None:
89+
def test_nnunetBundle(self) -> None:
9090
runner = nnUNetV2Runner(input_config=self.data_src_cfg, trainer_class_name="nnUNetTrainer_1epoch")
9191
with skip_if_downloading_fails():
9292
runner.run(run_train=False, run_find_best_configuration=False, run_predict_ensemble_postprocessing=False)
@@ -103,10 +103,7 @@ def test_nnunetBundle_get_trainer(self) -> None:
103103
print("LR Scheduler: ", nnunet_trainer.lr_scheduler)
104104
print("Device: ", nnunet_trainer.device)
105105
runner.train("3d_fullres")
106-
@skip_if_no_cuda
107-
def test_nnunetBundle_convert_bundle(self) -> None:
108106

109-
110107
nnunet_config = {
111108
"dataset_name_or_id": "001",
112109
"nnunet_trainer": "nnUNetTrainer_1epoch",
@@ -116,8 +113,6 @@ def test_nnunetBundle_convert_bundle(self) -> None:
116113
Path(self.bundle_root).joinpath("models").mkdir(parents=True, exist_ok=True)
117114
convert_nnunet_to_monai_bundle(nnunet_config, self.bundle_root, 0)
118115

119-
120-
def test_nnunetBundle_predict_from_bundle(self) -> None:
121116
data_transforms = Compose([
122117
LoadImaged(keys="image"),
123118
EnsureChannelFirstd(keys="image"),

0 commit comments

Comments
 (0)