@@ -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