From 1d1ba0bfc8234b56423a9a00abe1c7ccb7850acb Mon Sep 17 00:00:00 2001
From: Daniel Shields <Daniel.Shields@twosigma.com>
Date: Fri, 9 Feb 2024 11:20:26 -0600
Subject: [PATCH] update

---
 src/uberjob/_util/traceback.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/uberjob/_util/traceback.py b/src/uberjob/_util/traceback.py
index 5e7a1f4..9c7d641 100644
--- a/src/uberjob/_util/traceback.py
+++ b/src/uberjob/_util/traceback.py
@@ -48,10 +48,18 @@ def __repr__(self):
         )
 
 
+TruncatedStackFrame = None
+
+
 class TruncatedStackFrameType:
     def __repr__(self):
         return "TruncatedStackFrame"
 
+    def __new__(cls, *args, **kwargs):
+        if TruncatedStackFrame is not None:
+            return TruncatedStackFrame
+        return super().__new__(cls, *args, **kwargs)
+
 
 TruncatedStackFrame = TruncatedStackFrameType()
 
@@ -81,7 +89,7 @@ def recurse(frame, depth):
 def render_symbolic_traceback(stack_frame):
     stack_frames = []
     while stack_frame:
-        if type(stack_frame) is TruncatedStackFrameType:
+        if stack_frame is TruncatedStackFrame:
             stack_frames.append(stack_frame)
             break
         if "/IPython/core/" in stack_frame.path:
@@ -90,7 +98,7 @@ def render_symbolic_traceback(stack_frame):
         stack_frame = stack_frame.outer
 
     def format_stack_frame(s):
-        if type(s) is TruncatedStackFrameType:
+        if s is TruncatedStackFrame:
             return "  ... truncated"
         return f'  File "{s.path}", line {s.line}, in {s.name}'