Update nipype/interfaces/ants/visualization.py

effigies · web-flow · commit a5853e9c57ff · 2022-04-02T21:28:57.000-04:00
diff --git a/nipype/interfaces/ants/visualization.py b/nipype/interfaces/ants/visualization.py
@@ -28,8 +28,8 @@ class ConvertScalarImageToRGBInputSpec(ANTSCommandInputSpec):
     output_image = traits.Str(
         "rgb.nii.gz", argstr="%s", usedefault=True, desc="rgb output image", position=2
     )
-    mask_image = traits.Str(
-        "none", argstr="%s", desc="mask image", position=3, usedefault=True
+    mask_image = traits.Either(
+        "none", traits.File(exists=True), argstr="%s", desc="mask image", position=3, usedefault=True
     )
     colormap = traits.Enum(
         "grey",

Original file line number	Diff line number	Diff line change
`@@ -28,8 +28,8 @@ class ConvertScalarImageToRGBInputSpec(ANTSCommandInputSpec):`
`28`	`28`	`output_image = traits.Str(`
`29`	`29`	`"rgb.nii.gz", argstr="%s", usedefault=True, desc="rgb output image", position=2`
`30`	`30`	`)`
`31`		`- mask_image = traits.Str(`
`32`		`- "none", argstr="%s", desc="mask image", position=3, usedefault=True`
	`31`	`+ mask_image = traits.Either(`
	`32`	`+ "none", traits.File(exists=True), argstr="%s", desc="mask image", position=3, usedefault=True`
`33`	`33`	`)`
`34`	`34`	`colormap = traits.Enum(`
`35`	`35`	`"grey",`