Removed memory profiler stuff for now

Author: pintohutch

nipype/interfaces/base.py

@@ -1207,13 +1207,6 @@ def run_command(runtime, output=None, timeout=0.01, redirect_x=False):
     The returned runtime contains a merged stdout+stderr log with timestamps
     """
 
-    # Import packages
-    try:
-        from memory_profiler import _get_memory
-        mem_prof = True
-    except:
-        mem_prof = False
-
     # Init variables
     PIPE = subprocess.PIPE
     cmdline = runtime.cmdline

nipype/interfaces/utility.py

@@ -442,12 +442,7 @@ def _run_interface(self, runtime):
             if isdefined(value):
                 args[name] = value
 
-        # mem stuff
-        import memory_profiler
-        proc = (function_handle, (), args)
-        mem_mb, out = memory_profiler.memory_usage(proc=proc, retval=True, include_children=True, max_usage=True)
-        setattr(runtime, 'real_memory2', mem_mb[0]/1024.0)
-        #out = function_handle(**args)
+        out = function_handle(**args)
 
         if len(self._output_names) == 1:
             self._out[self._output_names[0]] = out

nipype/pipeline/plugins/multiproc.py

@@ -16,27 +16,11 @@
 def run_node(node, updatehash, plugin_args=None):
     result = dict(result=None, traceback=None)
     try:
-        run_memory = plugin_args['memory_profile']
-    except Exception:
-        run_memory = False
-    if run_memory:
-        import memory_profiler
-        import datetime
-        proc = (node.run, (), {'updatehash' : updatehash})
-        start = datetime.datetime.now()
-        mem_mb, retval = memory_profiler.memory_usage(proc=proc, retval=True, include_children=True, max_usage=True)
-        runtime = (datetime.datetime.now() - start).total_seconds()
-        result['result'] = retval
-        result['real_memory'] = mem_mb[0]/1024.0
-        result['real_memory2'] = retval.runtime.get('real_memory2')
-        result['run_seconds'] = runtime
-    else:
-        try:
-            result['result'] = node.run(updatehash=updatehash)
-        except:
-            etype, eval, etr = sys.exc_info()
-            result['traceback'] = format_exception(etype,eval,etr)
-            result['result'] = node.result
+        result['result'] = node.run(updatehash=updatehash)
+    except:
+        etype, eval, etr = sys.exc_info()
+        result['traceback'] = format_exception(etype,eval,etr)
+        result['result'] = node.result
     return result