first assert on peaks and overlays

karanphil · Apr 17, 2024 · 2eb2f38 · 2eb2f38
1 parent 3664a08
commit 2eb2f38
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/scripts/scil_viz_volume_screenshot.py b/scripts/scil_viz_volume_screenshot.py
@@ -97,13 +97,11 @@ def _parse_args(parser):
     args = parser.parse_args()
 
     assert_inputs_exist(parser, [args.volume],
-                        [args.overlays, args.labelmap, args.peaks])
+                        [args.labelmap] + args.overlays + args.peaks)
     assert_headers_compatible(parser, [args.volume],
-                              [args.overlays, args.labelmap, args.peaks])
+                              [args.labelmap] + args.overlays + args.peaks)
     assert_overlay_colors(args.overlays_colors, args.overlays, parser)
 
-    # TODO : check outputs (we need to know the slicing), could be glob
-
     return args
 
 

diff --git a/scripts/scil_viz_volume_screenshot_mosaic.py b/scripts/scil_viz_volume_screenshot_mosaic.py
@@ -111,10 +111,10 @@ def _parse_args(parser):
 
     output.append(args.out_fname)
 
-    assert_inputs_exist(parser, [args.volume], [args.overlays, args.labelmap])
+    assert_inputs_exist(parser, [args.volume], [args.labelmap] + args.overlays)
     assert_outputs_exist(parser, args, output)
     assert_headers_compatible(parser, [args.volume],
-                              [args.overlays, args.labelmap])
+                              [args.labelmap] + args.overlays)
     assert_overlay_colors(args.overlays_colors, args.overlays, parser)
 
     return args