Fix for issue #101

Author: fabianp

memory_profiler.py

@@ -51,8 +51,13 @@ def _get_memory(pid, timestamps=False, include_children=False):
             meminfo_attr = 'memory_info' if hasattr(process, 'memory_info') else 'get_memory_info'
             mem = getattr(process, meminfo_attr)()[0] / _TWO_20
             if include_children:
-                for p in process.get_children(recursive=True):
-                    mem += getattr(process, meminfo_attr)()[0] / _TWO_20
+                try:
+                    for p in process.get_children(recursive=True):
+                        mem += getattr(process, meminfo_attr)()[0] / _TWO_20
+                except AttributeError:
+                    # fix for newer psutil
+                    for p in process.children(recursive=True):
+                        mem += getattr(process, meminfo_attr)()[0] / _TWO_20
             if timestamps:
                 return (mem, time.time())
             else: