Mark typechecking frame as hidden.

patrick-kidger · Dec 22, 2024 · b2f16e8 · b2f16e8
1 parent 43059ff
commit b2f16e8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jaxtyping/_decorator.py b/jaxtyping/_decorator.py
@@ -440,6 +440,7 @@ def wrapped_fn(*args, **kwargs):  # pyright: ignore
             param_fn = _apply_typechecker(typechecker, param_fn)
 
             def wrapped_fn_impl(args, kwargs, bound, memos):
+                __tracebackhide__ = True
                 # First type-check just the parameters before the function is
                 # called.
                 try: