Remove main method

lungtumormask/dataprocessing.py

@@ -247,19 +247,4 @@ def post_process(left_mask, right_mask, preprocess_dump):
 
     stitched = np.logical_or(left, right).astype(int)
 
-    return stitched
-
-
-if __name__ == "__main__":
-    path = "D:\\Datasets\MSD\\Images\\lung_003.nii.gz"
-    preprocess_dump = preprocess(path)
-
-    unpad = post_process(preprocess_dump['left_lung'], preprocess_dump['right_lung'], preprocess_dump)
-
-    import nibabel
-
-    nimage = nibabel.Nifti1Image(unpad, preprocess_dump['org_affine'])
-    nibabel.save(nimage, "D:\\Datasets\\stitched.nii.gz")
-    
-
-    
+    return stitched

lungtumormask/mask.py

@@ -28,7 +28,4 @@ def mask(image_path, save_path):
 
     print(f"Storing segmentation at {save_path}")
     nimage = nibabel.Nifti1Image(infered, preprocess_dump['org_affine'])
-    nibabel.save(nimage, save_path)
-
-if __name__ == "__main__":
-    mask("D:\\Datasets\\MSD\\Images\\lung_003.nii.gz", "D:\\Datasets\\MSD\\Images\\out3.nii.gz")
+    nibabel.save(nimage, save_path)