fix: memory profiler psutil api updates

Lewiscowles1986 · Mar 31, 2024 · 542b561 · 542b561
1 parent cc6cc58
commit 542b561
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -155,7 +155,15 @@ jobs:
         echo ""
         echo ""
         echo "#"
-        echo "# Basic using example with experimental memory tracking"
+        echo "# Basic using example with experimental memory tracking (without psutil)"
+        echo "#"
+        python scripts/pycallgraph --memory graphviz -- examples/graphviz/basic.py
+        echo ""
+        echo ""
+        echo ""
+        pip install psutil
+        echo "#"
+        echo "# Basic using example with experimental memory tracking (with psutil)"
         echo "#"
         python scripts/pycallgraph --memory graphviz -- examples/graphviz/basic.py
         echo ""

diff --git a/pycallgraph/memory_profiler.py b/pycallgraph/memory_profiler.py
@@ -23,7 +23,7 @@
     def _get_memory(pid):
         process = psutil.Process(pid)
         try:
-            mem = float(process.get_memory_info()[0]) / (1024 ** 2)
+            mem = float(process.memory_info()[0]) / (1024 ** 2)
         except psutil.AccessDenied:
             mem = -1
         return mem