Merge pull request #10 from CBICA/ag-dev

small futureproofing for argparse
CBICA · Dec 6, 2024 · 5423f4d · 5423f4d
2 parents 2642dda + 61e8237
commit 5423f4d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/DLICV/__main__.py b/DLICV/__main__.py
@@ -53,7 +53,7 @@ def main() -> None:
     )
     parser.add_argument(
         "-o",
-        "out_dir",
+        "--out_dir",
         type=str,
         required=True,
         help="[REQUIRED] Output folder. If it does not exist it will be created. Predicted segmentations will have the same name as their source images.",
@@ -204,6 +204,8 @@ def main() -> None:
 
     args = parser.parse_args()
     args.f = [0]
+    args.i = args.in_dir
+    args.o = args.out_dir
 
     if args.clear_cache:
         shutil.rmtree(os.path.join(Path(__file__).parent, "nnunet_results"))