reduced symbolic traceback depth for performance reasons

twosigma · Oct 11, 2023 · 99427aa · 99427aa
1 parent 17fc24d
commit 99427aa
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/uberjob/_util/traceback.py b/src/uberjob/_util/traceback.py
@@ -56,7 +56,7 @@ def __repr__(self):
 TruncatedStackFrame = TruncatedStackFrameType()
 
 
-MAX_TRACEBACK_DEPTH = 15
+MAX_TRACEBACK_DEPTH = 3
 
 
 def get_stack_frame(initial_depth=2):
@@ -73,9 +73,8 @@ def recurse(frame, depth):
         )
 
     initial_frame = inspect.currentframe()
-    while initial_depth:
+    for _ in range(initial_depth):
         initial_frame = initial_frame.f_back
-        initial_depth -= 1
     return recurse(initial_frame, MAX_TRACEBACK_DEPTH)