diff --git a/dipy/workflows/base.py b/dipy/workflows/base.py index 8c7c1002bb..23f23b00e3 100644 --- a/dipy/workflows/base.py +++ b/dipy/workflows/base.py @@ -101,7 +101,7 @@ def add_workflow(self, workflow): ref_text = [text if text else "\n" for text in npds['References']] ref_idx = self.epilog.find('References: \n') + len('References: \n') self.epilog = "{0}{1}\n{2}".format(self.epilog[:ref_idx], - ''.join([text for text in ref_text]), + ''.join(ref_text), self.epilog[ref_idx:]) self.outputs = [param for param in npds['Parameters'] if @@ -109,11 +109,17 @@ def add_workflow(self, workflow): args, defaults = get_args_default(workflow.run) + output_args = self.add_argument_group('output arguments(optional)') + len_args = len(args) len_defaults = len(defaults) - output_args = \ - self.add_argument_group('output arguments(optional)') + if len_args != len(self.doc): + raise ValueError( + self.prog + ": Number of parameters in the " + "doc string and run method does not match. " + "Please ensure that the number of parameters " + "in the run method is same as the doc string.") for i, arg in enumerate(args): prefix = ''