fixes #14901 bug, made it theme dependent

GSAUC3 · Carreau · commit f63b4083918b · 2026-04-08T11:52:13.000+02:00
diff --git a/IPython/core/ultratb.py b/IPython/core/ultratb.py
@@ -450,7 +450,7 @@ class VerboseTB(TBTools):
     traceback, to be used with alternate interpreters (because their own code
     would appear in the traceback)."""
 
-    tb_highlight = "ansiblack bg:ansiyellow"
+    tb_highlight = "bg:ansiyellow"
     tb_highlight_style = "default"
 
     _mode: str
@@ -519,15 +519,6 @@ def __init__(
 
         self.skip_hidden = True
 
-        theme_highlight_map = {
-            "linux": "ansiblack bg:ansiyellow",
-            "neutral": "ansiblack bg:ansiyellow", 
-            "lightbg": "bg:ansiblue",
-            "nocolor": "", 
-        }
-
-        self.tb_highlight = theme_highlight_map.get(self._theme_name, "bg:ansiyellow")
-
     def format_record(self, frame_info: FrameInfo) -> str:
         """Format a single stack frame"""
         assert isinstance(frame_info, FrameInfo)
@@ -826,8 +817,10 @@ def get_records(self, etb: TracebackType, context: int, tb_offset: int) -> Any:
         after = context // 2
         before = context - after
         if self.has_colors:
-            base_style = theme_table[self._theme_name].as_pygments_style()
-            style = stack_data.style_with_executing_node(base_style, self.tb_highlight)
+            theme = theme_table[self._theme_name]
+            base_style = theme.as_pygments_style()
+            tb_highlight = theme.extra_style.get(Token.TbHighlight, self.tb_highlight)
+            style = stack_data.style_with_executing_node(base_style, tb_highlight)
             formatter = Terminal256Formatter(style=style)
         else:
             formatter = None
diff --git a/IPython/utils/PyColorize.py b/IPython/utils/PyColorize.py
@@ -138,6 +138,7 @@ def make_arrow(self, width: int) -> str:
         Token.PromptNum: "ansigreen bold",
         Token.OutPrompt: "ansibrightred",
         Token.OutPromptNum: "ansired bold",
+        Token.TbHighlight: "ansiblack bg:ansiyellow",
     },
 )
 
@@ -170,6 +171,7 @@ def make_arrow(self, width: int) -> str:
     Token.Name.Variable.Magic: "ansiblue",
     Token.Prompt: "ansigreen",
     Token.OutPrompt: "ansired",
+    Token.TbHighlight: "ansiblack bg:ansiyellow",
 }
 
 
@@ -227,6 +229,7 @@ def make_arrow(self, width: int) -> str:
         Token.PromptNum: "ansiblue bold",
         Token.OutPrompt: "ansibrightred",
         Token.OutPromptNum: "ansired bold",
+        Token.TbHighlight: "ansired bg:ansiyellow",
     },
 )
 
@@ -268,6 +271,7 @@ def make_arrow(self, width: int) -> str:
         Token.PromptNum: "ansiblue bold",
         Token.OutPrompt: "ansibrightred",
         Token.OutPromptNum: "ansired bold",
+        Token.TbHighlight: f"bg:{PRIDE_YELLOW}",
     },
     symbols={"arrow_body": "\u2500", "arrow_head": "\u25b6", "top_line": "\u2500"},
 )
@@ -368,6 +372,7 @@ def make_arrow(self, width: int) -> str:
         Token.PromptNum: f"{GRUVBOX_PROMPT_NUM} bold",
         Token.OutPrompt: GRUVBOX_OUT_PROMPT,
         Token.OutPromptNum: f"{GRUVBOX_OUT_PROMPT_NUM} bold",
+        Token.TbHighlight: f"bg:{GRUVBOX_TOPLINE}",
     },
     symbols={"arrow_body": "\u2500", "arrow_head": "\u25b6", "top_line": "\u2500"},
 )
