use pss memory info type if available

echarles · echarles · commit 14c385318201 · 2022-06-13T17:44:27.000+02:00
diff --git a/ipykernel/kernelbase.py b/ipykernel/kernelbase.py
@@ -982,9 +982,10 @@ async def usage_request(self, stream, ident, parent):
                 for process in self.processes.values()
             ]
         )
+        mem_info_type = "pss" if hasattr(current_process.memory_full_info(), "pss") else "rss"
         reply_content["kernel_memory"] = sum(
             [
-                self.get_process_metric_value(process, "memory_info", "rss")
+                self.get_process_metric_value(process, "memory_full_info", mem_info_type)
                 for process in self.processes.values()
             ]
         )

Original file line number	Diff line number	Diff line change
`@@ -982,9 +982,10 @@ async def usage_request(self, stream, ident, parent):`
`982`	`982`	`for process in self.processes.values()`
`983`	`983`	`]`
`984`	`984`	`)`
	`985`	`+ mem_info_type = "pss" if hasattr(current_process.memory_full_info(), "pss") else "rss"`
`985`	`986`	`reply_content["kernel_memory"] = sum(`
`986`	`987`	`[`
`987`		`- self.get_process_metric_value(process, "memory_info", "rss")`
	`988`	`+ self.get_process_metric_value(process, "memory_full_info", mem_info_type)`
`988`	`989`	`for process in self.processes.values()`
`989`	`990`	`]`
`990`	`991`	`)`