Added try-blocks around the runtime profile stats in callback logger

Author: pintohutch

nipype/pipeline/plugins/callback_log.py

@@ -2,22 +2,43 @@
 import logging
 
 def log_nodes_cb(node, status, result=None):
+    '''
+    '''
+
+    # Init variables
     logger = logging.getLogger('callback')
+
+    # Check runtime profile stats
     if result is None:
         node_mem = cmd_mem = run_seconds = cmd_threads = 'N/A'
     else:
-        node_mem = result['node_memory']
-        cmd_mem = result['cmd_memory']
-        run_seconds = result['run_seconds']
-        cmd_threads = result['cmd_threads']
+        try:
+            node_mem = result['node_memory']
+        except KeyError:
+            node_mem = 'Unknown'
+        try:
+            cmd_mem = result['cmd_memory']
+        except KeyError:
+            cmd_mem = 'Unknown'
+        try:
+            run_seconds = result['run_seconds']
+        except KeyError:
+            run_seconds = 'Unknown'
+        try:
+            cmd_threads = result['cmd_threads']
+        except:
+            cmd_threads = 'Unknown'
+
+    # Check status and write to log
+    # Start
     if status == 'start':
         message  = '{"name":' + '"' + node.name + '"' + ',"id":' + '"' +\
         node._id + '"' + ',"start":' + '"' +str(datetime.datetime.now()) +\
         '"' + ',"estimated_memory":' + str(node._interface.estimated_memory) + ',"num_threads":' \
         + str(node._interface.num_threads) + '}'
 
         logger.debug(message)
-
+    # End
     elif status == 'end':
         message  = '{"name":' + '"' + node.name + '"' + ',"id":' + '"' + \
         node._id + '"' + ',"finish":' + '"' + str(datetime.datetime.now()) +  \
@@ -29,7 +50,7 @@ def log_nodes_cb(node, status, result=None):
         ',"run_seconds":' + '"'+ str(run_seconds) + '"'+ '}'
 
         logger.debug(message)
-
+    # Other
     else:
         message  = '{"name":' + '"' + node.name + '"' + ',"id":' + '"' + \
         node._id + '"' + ',"finish":' + '"' + str(datetime.datetime.now()) +\

nipype/pipeline/plugins/tests/test_multiproc.py

@@ -212,7 +212,8 @@ def test_do_not_use_more_threads_then_specified():
     pipe.connect(n3, 'output1', n4, 'input2')
     n1.inputs.input1 = 10
     pipe.config['execution']['poll_sleep_duration'] = 1
-    pipe.run(plugin='ResourceMultiProc', plugin_args={'n_procs': max_threads, 'status_callback': log_nodes_cb})
+    pipe.run(plugin='ResourceMultiProc', plugin_args={'n_procs': max_threads,
+                                                      'status_callback': log_nodes_cb})
 
     nodes, last_node = draw_gantt_chart.log_to_json(LOG_FILENAME)
     #usage in every second