FIX: Annotate type, add environment argument to _run_task

effigies · effigies · commit ee5c86e35729 · 2024-03-22T21:25:54.000-04:00
diff --git a/pydra/tasks/nipype1/utils.py b/pydra/tasks/nipype1/utils.py
@@ -11,7 +11,7 @@ def traitedspec_to_specinfo(traitedspec):
     return pydra.specs.SpecInfo(
         name="Inputs",
         fields=[
-            (name, attrs.field(metadata={"help_string": trait.desc}))
+            (name, attrs.field(metadata={"help_string": trait.desc}, type=ty.Any))
             for name, trait in traitedspec.traits().items()
             if name in trait_names
         ],
@@ -68,7 +68,7 @@ def __init__(
         )
         self.output_spec = traitedspec_to_specinfo(interface._outputs())
 
-    def _run_task(self):
+    def _run_task(self, environment=None):
         inputs = attrs.asdict(self.inputs, filter=lambda a, v: v is not attrs.NOTHING)
         node = nipype.Node(self._interface, base_dir=self.output_dir, name=self.name)
         node.inputs.trait_set(**inputs)
