Fix handling of missing parent header in record activity (#1498)

jupyter-server · Feb 12, 2025 · 641e8fc · 641e8fc
1 parent 5ef77a3
commit 641e8fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jupyter_server/services/kernels/kernelmanager.py b/jupyter_server/services/kernels/kernelmanager.py
@@ -597,7 +597,8 @@ def record_activity(msg_list):
             msg = session.deserialize(fed_msg_list, content=False)
 
             msg_type = msg["header"]["msg_type"]
-            parent_msg_type = msg.get("parent_header", {}).get("msg_type", None)
+            parent_header = msg.get("parent_header")
+            parent_msg_type = None if parent_header is None else parent_header.get("msg_type")
             if (
                 self.track_message_type(msg_type)
                 or self.track_message_type(parent_msg_type)