actual actual bugfix for tracing reenablement in module stmt

ipyflow · Dec 6, 2023 · 79525de · 79525de
1 parent 72ca706
commit 79525de
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/core/ipyflow/tracing/ipyflow_tracer.py b/core/ipyflow/tracing/ipyflow_tracer.py
@@ -1480,6 +1480,9 @@ def _try_reenable_tracing(
         )
         if not is_module_stmt and tracing_reenabled_call_stack_length == -1:
             return False
+        tracing_reenabled_call_stack_length = max(
+            tracing_reenabled_call_stack_length, 0
+        )
         assert tracing_reenabled_call_stack_length <= len(self.call_stack)
         while len(self.call_stack) > tracing_reenabled_call_stack_length:
             self.call_stack.pop()