Merge pull request jax-ml#23726 from hawkinsp:debug

PiperOrigin-RevId: 676030839
keshavb96 · Sep 18, 2024 · cd04d0f · cd04d0f
2 parents 016c499 + c756d9b
commit cd04d0f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/jax/_src/debugger/core.py b/jax/_src/debugger/core.py
@@ -112,6 +112,11 @@ def from_frameinfo(cls, frame_info) -> DebuggerFrame:
       # then we subtract it off from the `lineno` and don't need to subtract 1
       # since both start and lineno are 1-indexed.
       offset = frame_info.lineno - max(start, 1)
+      if offset >= len(source):
+        # Sometimes we don't get a valid source/offset pair. This seems to
+        # happen sometimes when code uses eval(). If that happens, give up.
+        source = []
+        offset = None
     except OSError:
       source = []
       offset = None