From b2f16e84ee635cec52db2c1b766e533db620c5b7 Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Sun, 22 Dec 2024 01:58:17 +0100 Subject: [PATCH] Mark typechecking frame as hidden. --- jaxtyping/_decorator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/jaxtyping/_decorator.py b/jaxtyping/_decorator.py index ef876a5..fadbb37 100644 --- 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: