Skip to content

Commit 8d8adeb

Browse files
committed
ENH: Use attrs namespace
1 parent 75b72f2 commit 8d8adeb

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

pydra/tasks/nipype1/utils.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pydra
22
import nipype
3-
import attr
3+
import attrs
44
import typing as ty
55

66

@@ -9,7 +9,7 @@ def traitedspec_to_specinfo(traitedspec):
99
return pydra.specs.SpecInfo(
1010
name="Inputs",
1111
fields=[
12-
(name, attr.ib(type=ty.Any, metadata={"help_string": trait.desc}))
12+
(name, attrs.field(metadata={"help_string": trait.desc}))
1313
for name, trait in traitedspec.traits().items()
1414
if name in trait_names
1515
],
@@ -67,9 +67,7 @@ def __init__(
6767
self.output_spec = traitedspec_to_specinfo(interface._outputs())
6868

6969
def _run_task(self):
70-
inputs = attr.asdict(
71-
self.inputs, filter=lambda a, v: v is not attr.NOTHING, retain_collection_types=True
72-
)
70+
inputs = attrs.asdict(self.inputs, filter=lambda a, v: v is not attrs.NOTHING)
7371
node = nipype.Node(self._interface, base_dir=self.output_dir, name=self.name)
7472
node.inputs.trait_set(**inputs)
7573
res = node.run()

0 commit comments

Comments
 (0)