FIX: improve evaluate_connect_function error handling.

This incorporates Chris's suggestions from #3655 (comment) Except I raise a new error and include the original error message (instead of revising the original error message). Co-authored-by: Chris Markiewicz <[email protected]>
nipy · Oct 10, 2024 · df241b2 · df241b2
1 parent 469d0c6
commit df241b2
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/nipype/pipeline/engine/utils.py b/nipype/pipeline/engine/utils.py
@@ -689,15 +689,10 @@ def evaluate_connect_function(function_source, args, first_arg):
     try:
         output_value = func(first_arg, *list(args))
     except NameError as e:
-        if e.args[0].endswith("is not defined"):
-            e.args = (
-                e.args[0],
-                (
-                    "Due to engine constraints all imports have to be done "
-                    "inside each function definition"
-                ),
-            )
-        raise e
+        raise NameError(
+            f"{e}: Due to engine constraints all imports have to be done inside each "
+            " function definition."
+        )
     return output_value