vh#|UddlmZddlmZddlZddlZddlZddlZddlZddl m Z m Z m Z ddl mZddl mZddlmZe ded e f Zeej&gZd ed <d#d ZdZd$dZd%dZd&dZd%dZd'dZd(dZd)dZd*dZd+dZ d,dZ!Gdde"Z#dZ$Gdde"Z%de%_&d-dZ'd-d Z(d.d!Z)d/d"Z*y)0) annotations)CallableN)AnyTypeVarcast)config)util)_jaxC.)boundz list[str]_exclude_pathsc.tj|yN)r append)paths R/opt/face_recognition/venv/lib/python3.12/site-packages/jax/_src/traceback_util.pyregister_exclusionr"szThe stack trace below excludes JAX-internal frames. The preceding is the original exception that occurred, unmodified. --------------------cNtjj|}tjj|} tjj||g} ||k(xs tjj ||S#t$rYywxYw#t $rYywxYw)NF)osrabspath commonpath ValueErrorsamefileOSError)r path_prefixcommons r_path_starts_withr*s  $ ,+ WW  { 3 4F  [ IBGG$4$4V[$II     s$!B "&B BB B$#B$c@t|jjSr)include_filenamef_code co_filenamefs r include_framer%:s !((.. //rc6tfdtD S)Nc36K|]}t|ywr)r).0rfilenames r z#include_filename..>sNt"8T2Ns)anyr )r)s`rr r =s N~N N NNrc2d|jjvS)Nzimportlib._bootstrap)r!r"r#s r_ignore_known_hidden_framer-Cs 188#7#7 77rcttj|D] \}}t|rd|jd<"y)NT__tracebackhide__) tracebackwalk_tbr%f_locals)tbr$_linenos r#_add_tracebackhide_to_hidden_framesr5Fs7%%b)-ja  (,ajj$%-rcd}ttj|}t|D]3\}}t |st j |||j|}5|Sr)listr0r1reversedr%types TracebackTypef_lasti)r3outframesr$linenos rfilter_tracebackr?Ks] #  !!"% &&F#;iaQ   Q 6 :c; *rcB|}d}tj|jD]x\}}t|r|r|jj dk7r|St |r"tj|||j|}|jj dk(swd}z|S)NFzT) r0 walk_stacktb_framer-r!co_namer%r9r:r;)r3r<reached_module_levelr$r>s r_add_call_stack_framesrETs #'' 4"ia!!$ 0 0J > * Q   Q 6 :cxx:%!" *rcJ|jtk(xr|jdk(S)Nreraise_with_filtered_traceback)r)__file__namer#s r_is_reraiser_framerJls% **  6 &&5 57rc|jytj|jj}t d|ddDS)NFc32K|]}t|ywr)rJ)r(r$s rr*z%_is_under_reraiser..ts 4q  " 4s) __traceback__r0 extract_stackrBr+)er3s r_is_under_reraiserrQps@__ q778" 4BsG 4 44rczdjtjt||j S)N)joinr0format_exception_onlytypestrip)rPs rrUrUvs+ 00a!< = C C EErc eZdZy)UnfilteredStackTraceN)__name__ __module__ __qualname__rrrYrYysrrYzFor simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.ceZdZdZy)SimplifiedTracebackctSr)_simplified_tb_msg)selfs r__str__zSimplifiedTraceback.__str__s rN)rZr[r\rcr]rrr_r_srr_z jax.errorsc8 ty#t$rYywxYw)z6Returns true if we appear to be in an IPython session.TF) get_ipython NameErrorr]rr_running_under_ipythonrgs!M   s c.ddl}|jdddk\S)z?Returns true if the IPython version supports __tracebackhide__.rN))IPython version_info)rls r_ipython_supports_tracebackhiderns   bq !W ,,rc|tjj}||dk(rtrt rd}|Sd}|S)Nauto tracebackhidequiet_remove_frames)rtraceback_filteringvaluergrn)modes r_filtering_modervsB  # # ) )$ \TV^ %D%F d +#d +rc^tjfd}tt|S)a Wraps ``fun`` to form a boundary for filtering exception tracebacks. When an exception occurs below ``fun``, this appends to it a custom ``__cause__`` that carries a filtered traceback. The traceback imitates the stack trace of the original exception, but with JAX-internal frames removed. This boundary annotation works in composition with itself. The topmost frame corresponding to an :func:`~api_boundary` is the one below which stack traces are filtered. In other words, if ``api_boundary(f)`` calls ``api_boundary(g)``, directly or indirectly, the filtered stack trace provided is the same as if ``api_boundary(f)`` were to simply call ``g`` instead. This annotation is primarily useful in wrapping functions output by JAX's transformations. For example, consider ``g = jax.jit(f)``. When ``g`` is called, JAX's JIT compilation machinery is invoked, which in turn calls ``f`` in order to trace and translate it. If the function ``f`` raises an exception, the stack unwinds through JAX's JIT internals up to the original call site of ``g``. Because the function returned by :func:`~jax.jit` is annotated as an :func:`~api_boundary`, such an exception is accompanied by an additional traceback that excludes the frames specific to JAX's implementation. cd}  |i|S#t$ri}t}t|s|dk(r|dk(rt|jd\}} |j}t |}|j |ttdrtj|tjdk\r|dk(r|jdtz|dk(r t}nP|d k(r.reraise_with_filtered_tracebacks1 $ !& !! /  d A $%-  +AOO< *k:& __&r*  % 47 8  + +K 8   w &43H+H **-0BB C$ !* *,-I&'*CE123C,S1I  $ $%;B%? @7v=STU U ! ) "#--) +,+A+A) (!!+!-   _/s' F;E;D E33E88E;;F) functoolswrapsrr )rrGs` r api_boundaryrs1. ??333h a0 11r)rstr)rrrrreturnbool)r$ztypes.FrameTyperr)r)rrr)r3types.TracebackType)r3rrztypes.TracebackType | None)r3rrr)r$ztraceback.FrameSummaryrr)rP BaseExceptionrr)rPrrr)rr)rr)rr rr )+ __future__rcollections.abcrrrrr0r9typingrrrjax._srcrr jax._src.libr r rHr __annotations__rrrr%r r-r5r?rErJrQrUr~rYrar_r[rgrnrvrr]rrrs#$  %% CxS)*%t}}5 5  0O 8-   075 F,9+F)".- L2r