Replace deprecated HasTraits.set with trait_set

Author: effigies

nipype/caching/memory.py

@@ -70,7 +70,7 @@ def __call__(self, **kwargs):
         kwargs = modify_paths(kwargs, relative=False)
         interface = self.interface()
         # Set the inputs early to get some argument checking
-        interface.inputs.set(**kwargs)
+        interface.inputs.trait_set(**kwargs)
         # Make a name for our node
         inputs = interface.inputs.get_hashval()
         hasher = hashlib.new('md5')

nipype/interfaces/afni/utils.py

@@ -171,9 +171,7 @@ def aggregate_outputs(self, runtime=None, needed_outputs=None):
             m = re.search(pattern, line)
             if m:
                 d = m.groupdict()
-                for k in list(d.keys()):
-                    d[k] = int(d[k])
-                outputs.set(**d)
+                outputs.trait_set(**{k, int(d[k]) for k in d.keys()})
         return outputs
 
 

nipype/interfaces/base.py

@@ -367,7 +367,7 @@ def __init__(self, **kwargs):
                 undefined_traits[trait] = Undefined
         self.trait_set(trait_change_notify=False, **undefined_traits)
         self._generate_handlers()
-        self.set(**kwargs)
+        self.trait_set(**kwargs)
 
     def items(self):
         """ Name, trait generator for user modifiable traits
@@ -650,7 +650,7 @@ def __deepcopy__(self, memo):
                 pass
         # clone twice
         dup = self.clone_traits(memo=memo)
-        dup.set(**dup_dict)
+        dup.trait_set(**dup_dict)
         return dup
 
 
@@ -1060,7 +1060,7 @@ def run(self, **inputs):
         results :  an InterfaceResult object containing a copy of the instance
         that was executed, provenance information and, if successful, results
         """
-        self.inputs.set(**inputs)
+        self.inputs.trait_set(**inputs)
         self._check_mandatory_inputs()
         self._check_version_requirements(self.inputs)
         interface = self.__class__

nipype/interfaces/utility/base.py

@@ -72,7 +72,7 @@ def __init__(self, fields=None, mandatory_inputs=True, **inputs):
         # Adding any traits wipes out all input values set in superclass initialization,
         # even it the trait is not in the add_traits argument. The work-around is to reset
         # the values after adding the traits.
-        self.inputs.set(**inputs)
+        self.inputs.trait_set(**inputs)
 
     def _add_output_traits(self, base):
         return add_traits(base, self._fields)
@@ -319,7 +319,7 @@ class Split(IOBase):
 
     >>> from nipype.interfaces.utility import Split
     >>> sp = Split()
-    >>> _ = sp.inputs.set(inlist=[1, 2, 3], splits=[2, 1])
+    >>> _ = sp.inputs.trait_set(inlist=[1, 2, 3], splits=[2, 1])
     >>> out = sp.run()
     >>> out.outputs.out1
     [1, 2]
@@ -373,12 +373,12 @@ class Select(IOBase):
 
     >>> from nipype.interfaces.utility import Select
     >>> sl = Select()
-    >>> _ = sl.inputs.set(inlist=[1, 2, 3, 4, 5], index=[3])
+    >>> _ = sl.inputs.trait_set(inlist=[1, 2, 3, 4, 5], index=[3])
     >>> out = sl.run()
     >>> out.outputs.out
     4
 
-    >>> _ = sl.inputs.set(inlist=[1, 2, 3, 4, 5], index=[3, 4])
+    >>> _ = sl.inputs.trait_set(inlist=[1, 2, 3, 4, 5], index=[3, 4])
     >>> out = sl.run()
     >>> out.outputs.out
     [4, 5]

nipype/pipeline/engine/nodes.py

@@ -489,14 +489,14 @@ def _save_results(self, result, cwd):
                 outputs = result.outputs.get()
             except TypeError:
                 outputs = result.outputs.dictcopy()  # outputs was a bunch
-            result.outputs.set(**modify_paths(outputs, relative=True,
-                                              basedir=cwd))
+            result.outputs.trait_set(**modify_paths(outputs, relative=True,
+                                                    basedir=cwd))
 
         savepkl(resultsfile, result)
         logger.debug('saved results in %s', resultsfile)
 
         if result.outputs:
-            result.outputs.set(**outputs)
+            result.outputs.trait_set(**outputs)
 
     def _load_resultfile(self, cwd):
         """Load results if it exists in cwd
@@ -543,9 +543,9 @@ def _load_resultfile(self, cwd):
                     except TypeError:
                         outputs = result.outputs.dictcopy()  # outputs == Bunch
                     try:
-                        result.outputs.set(**modify_paths(outputs,
-                                                          relative=False,
-                                                          basedir=cwd))
+                        result.outputs.trait_set(**modify_paths(outputs,
+                                                                relative=False,
+                                                                basedir=cwd))
                     except FileNotFoundError:
                         logger.debug('conversion to full path results in '
                                      'non existent file')
@@ -561,7 +561,7 @@ def _load_results(self, cwd):
             logger.debug('aggregating results')
             if attribute_error:
                 old_inputs = loadpkl(op.join(cwd, '_inputs.pklz'))
-                self.inputs.set(**old_inputs)
+                self.inputs.trait_set(**old_inputs)
             if not isinstance(self, MapNode):
                 self._copyfiles_to_wd(cwd, True, linksonly=True)
                 aggouts = self._interface.aggregate_outputs(
@@ -1121,7 +1121,7 @@ def _make_nodes(self, cwd=None):
                         base_dir=op.join(cwd, 'mapflow'),
                         name=nodename)
             node.plugin_args = self.plugin_args
-            node._interface.inputs.set(
+            node._interface.inputs.trait_set(
                 **deepcopy(self._interface.inputs.get()))
             for field in self.iterfield:
                 if self.nested:
@@ -1246,7 +1246,7 @@ def _get_inputs(self):
         old_inputs = self._inputs.get()
         self._inputs = self._create_dynamic_traits(self._interface.inputs,
                                                    fields=self.iterfield)
-        self._inputs.set(**old_inputs)
+        self._inputs.trait_set(**old_inputs)
         super(MapNode, self)._get_inputs()
 
     def _check_iterfield(self):