Fix generate5tt input

teanijarv · web-flow · commit eda649755b40 · 2024-02-28T09:03:21.000+01:00
diff --git a/nipype/interfaces/mrtrix3/utils.py b/nipype/interfaces/mrtrix3/utils.py
@@ -11,6 +11,7 @@
     traits,
     TraitedSpec,
     File,
+    Directory,
     InputMultiPath,
     isdefined,
 )
@@ -230,8 +231,10 @@ class Generate5ttInputSpec(MRTrix3BaseInputSpec):
         mandatory=True,
         desc="tissue segmentation algorithm",
     )
-    in_file = File(
-        exists=False, argstr="%s", mandatory=True, position=-2, desc="input image"
+    in_file = traits.Either(
+        File(exists=True),
+        Directory(exists=True),
+        argstr="%s", mandatory=True, position=-2, desc="input image / directory"
     )
     out_file = File(argstr="%s", mandatory=True, position=-1, desc="output image")
 

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,7 @@`
`11`	`11`	`traits,`
`12`	`12`	`TraitedSpec,`
`13`	`13`	`File,`
	`14`	`+ Directory,`
`14`	`15`	`InputMultiPath,`
`15`	`16`	`isdefined,`
`16`	`17`	`)`
`@@ -230,8 +231,10 @@ class Generate5ttInputSpec(MRTrix3BaseInputSpec):`
`230`	`231`	`mandatory=True,`
`231`	`232`	`desc="tissue segmentation algorithm",`
`232`	`233`	`)`
`233`		`- in_file = File(`
`234`		`- exists=False, argstr="%s", mandatory=True, position=-2, desc="input image"`
	`234`	`+ in_file = traits.Either(`
	`235`	`+ File(exists=True),`
	`236`	`+ Directory(exists=True),`
	`237`	`+ argstr="%s", mandatory=True, position=-2, desc="input image / directory"`
`235`	`238`	`)`
`236`	`239`	`out_file = File(argstr="%s", mandatory=True, position=-1, desc="output image")`
`237`	`240`