Update pydra/compose/python.py

Co-authored-by: Chris Markiewicz <[email protected]>

Author: tclose

pydra/compose/python.py

@@ -236,11 +236,10 @@ def _run(self, job: "Job[Task]", rerun: bool = True) -> None:
         # Run the actual function
         returned = self.function(**inputs)
         # Collect the outputs and save them into the job.return_values dictionary
-        job.return_values = {f.name: f.default for f in task_fields(self.Outputs)}
-        return_names = list(job.return_values)
+        return_names = [f.name for f in task_fields(self.Outputs)]
         if returned is None:
             job.return_values = {nm: None for nm in return_names}
-        elif len(job.return_values) == 1:
+        elif len(return_names) == 1:
             # if only one element in the fields, everything should be returned together
             job.return_values = {list(job.return_values)[0]: returned}
         elif isinstance(returned, tuple) and len(return_names) == len(returned):