Format utils.py and init.py

CBICA · Nov 22, 2024 · 885eef9 · 885eef9
1 parent 9f78a42
commit 885eef9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/DLWMLS/__main__.py b/DLWMLS/__main__.py
@@ -246,7 +246,7 @@ def main() -> None:
     model_folder = os.path.join(
         Path(__file__).parent,
         "nnunet_results",
-        "Dataset004_Dataset003_qcedCorrected/nnUNetTrainer__nnUNetPlans__3d_fullres"
+        "Dataset004_Dataset003_qcedCorrected/nnUNetTrainer__nnUNetPlans__3d_fullres",
         # "Dataset%s_Dataset%s_dlwmlsICVAll/nnUNetTrainer__nnUNetPlans__%s"
         # "Dataset%s_Task%s_DLWMLSV2/nnUNetTrainer__nnUNetPlans__%s/"
         # % (args.d, args.d, args.c),

diff --git a/DLWMLS/utils.py b/DLWMLS/utils.py
@@ -37,6 +37,9 @@ def rename_and_copy_files(src_folder: str, des_folder: str) -> Tuple[dict, dict]
         shutil.copy2(old_name, new_name)
         rename_dict[filename] = rename_file
         rename_back_dict[rename_back] = filename.split(".nii")[0] + "_WMLS.nii.gz"
-        print("%s will be renamed to %s" % (old_name, filename.split(".nii")[0] + "_WMLS.nii.gz"))
+        print(
+            "%s will be renamed to %s"
+            % (old_name, filename.split(".nii")[0] + "_WMLS.nii.gz")
+        )
 
     return rename_dict, rename_back_dict