@@ -142,15 +142,15 @@ def write_report(node, report_type=None, is_mapnode=False):
142142 ['Hierarchy : %s' % node .fullname ,
143143 'Exec ID : %s' % node ._id ]),
144144 write_rst_header ('Original Inputs' , level = 1 ),
145- write_rst_dict (node .inputs .get ()),
145+ write_rst_dict (node .inputs .trait_get ()),
146146 ]
147147 with open (report_file , 'wt' ) as fp :
148148 fp .write ('\n ' .join (lines ))
149149 return
150150
151151 lines = [
152152 write_rst_header ('Execution Inputs' , level = 1 ),
153- write_rst_dict (node .inputs .get ()),
153+ write_rst_dict (node .inputs .trait_get ()),
154154 ]
155155
156156 result = node .result # Locally cache result
@@ -166,7 +166,7 @@ def write_report(node, report_type=None, is_mapnode=False):
166166 if isinstance (outputs , Bunch ):
167167 lines .append (write_rst_dict (outputs .dictcopy ()))
168168 elif outputs :
169- lines .append (write_rst_dict (outputs .get ()))
169+ lines .append (write_rst_dict (outputs .trait_get ()))
170170
171171 if is_mapnode :
172172 lines .append (write_rst_header ('Subnode reports' , level = 1 ))
@@ -238,8 +238,8 @@ def save_resultfile(result, cwd, name):
238238 resultsfile = os .path .join (cwd , 'result_%s.pklz' % name )
239239 if result .outputs :
240240 try :
241- outputs = result .outputs .get ()
242- except TypeError :
241+ outputs = result .outputs .trait_get ()
242+ except AttributeError :
243243 outputs = result .outputs .dictcopy () # outputs was a bunch
244244 result .outputs .set (** modify_paths (outputs , relative = True , basedir = cwd ))
245245
@@ -293,8 +293,8 @@ def load_resultfile(path, name):
293293 else :
294294 if result .outputs :
295295 try :
296- outputs = result .outputs .get ()
297- except TypeError :
296+ outputs = result .outputs .trait_get ()
297+ except AttributeError :
298298 outputs = result .outputs .dictcopy () # outputs == Bunch
299299 try :
300300 result .outputs .set (
@@ -1391,19 +1391,19 @@ def clean_working_directory(outputs,
13911391 """
13921392 if not outputs :
13931393 return
1394- outputs_to_keep = list (outputs .get ().keys ())
1394+ outputs_to_keep = list (outputs .trait_get ().keys ())
13951395 if needed_outputs and \
13961396 str2bool (config ['execution' ]['remove_unnecessary_outputs' ]):
13971397 outputs_to_keep = needed_outputs
13981398 # build a list of needed files
13991399 output_files = []
1400- outputdict = outputs .get ()
1400+ outputdict = outputs .trait_get ()
14011401 for output in outputs_to_keep :
14021402 output_files .extend (walk_outputs (outputdict [output ]))
14031403 needed_files = [path for path , type in output_files if type == 'f' ]
14041404 if str2bool (config ['execution' ]['keep_inputs' ]):
14051405 input_files = []
1406- inputdict = inputs .get ()
1406+ inputdict = inputs .trait_get ()
14071407 input_files .extend (walk_outputs (inputdict ))
14081408 needed_files += [path for path , type in input_files if type == 'f' ]
14091409 for extra in [
@@ -1435,7 +1435,7 @@ def clean_working_directory(outputs,
14351435 else :
14361436 if not str2bool (config ['execution' ]['keep_inputs' ]):
14371437 input_files = []
1438- inputdict = inputs .get ()
1438+ inputdict = inputs .trait_get ()
14391439 input_files .extend (walk_outputs (inputdict ))
14401440 input_files = [path for path , type in input_files if type == 'f' ]
14411441 for f in walk_files (cwd ):
0 commit comments