remove unused import

sct-pipeline · Jan 22, 2024 · e43165a · e43165a
1 parent 71bda2f
commit e43165a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/monai/run_inference_single_image.py b/monai/run_inference_single_image.py
@@ -16,7 +16,7 @@
 from time import time
 
 from monai.inferers import sliding_window_inference
-from monai.data import (DataLoader, Dataset, load_decathlon_datalist, decollate_batch)
+from monai.data import (DataLoader, Dataset, decollate_batch)
 from monai.transforms import (Compose, EnsureTyped, Invertd, SaveImage, Spacingd,
                               LoadImaged, NormalizeIntensityd, EnsureChannelFirstd, 
                               DivisiblePadd, Orientationd, ResizeWithPadOrCropd)
@@ -284,7 +284,7 @@ def main():
             # binarize the prediction with a threshold of 0.5
             pred[pred >= 0.5] = 1
             pred[pred < 0.5] = 0
-                            
+
             # get subject name
             subject_name = (batch["image_meta_dict"]["filename_or_obj"][0]).split("/")[-1].replace(".nii.gz", "")
             logger.info(f"Saving subject: {subject_name}")