o ={c@sdZddlZddlZddlZddlZddlZddlmmZ ej ej e ddej ddfZddZdd Zd d Zdd d ZddZdS)z=Utilities related to Keras exception stack trace prettifying.Nz..Z tensorflowZpythoncCstD] }||vr dSqdS)NFT)_EXCLUDED_PATHS)fnameZ exclusionr;lib/python3.10/site-packages/keras/utils/traceback_utils.py include_frames rcCspd}tt|}t|D]\}}t|jjr t|||j |}q |dur6|r6|d\}}t|||j |}|S)zFIterate through traceback frames and return a new, filtered traceback.N) list tracebackwalk_tbreversedrf_code co_filenametypes TracebackTypef_lasti)tbZlast_tbZtb_listfZline_norrr_process_traceback_frames&s   rcs8tjjdks tjjdkrSfdd}tjj|S)zLFilter out Keras-internal stack trace frames in exceptions raised by fn.c sftjs |i|Sd}z"z |i|WW~Sty0}z t|j}||dd}~ww~wN)tfZ debuggingZis_traceback_filtering_enabled Exceptionr __traceback__with_traceback)argskwargsZ filtered_tbefnrr error_handler;s   z'filter_traceback..error_handler)sys version_infomajorminorr __internal__ decoratormake_decorator)rr rrrfilter_traceback5s r(csfdd}tjj|S)aAdd information about call argument values to an error message. Arguments: fn: Function to wrap. Exceptions raised by the this function will be re-raised with additional information added to the error message, displaying the values of the different arguments that the function was called with. object_name: String, display name of the class/function being called, e.g. `'layer "layer_name" (LayerClass)'`. Returns: A wrapped version of `fn`. c sd}d}zz |i|WW~~Sty}zt|dr|t}z |j|i|}Wn ty7|wg}t|jD]#}|j |j vrUt j t|j |j }n|j}|d|j d|qA|rd|}t|t jjrw|j}n |jr|jd}nd}rnj} d| d|d | d |}t|t jjr||j|j||j} nz||} Wn tyt|} Ynwd | _n|} | |jdd}~ww~~w) N_keras_call_info_injectedu • = rz#Exception encountered when calling z. z Call arguments received by z: T) rhasattrinspect signatureZbind TypeErrorrZ parametersvaluesnameZ argumentsrZnestZ map_structureformat_argument_valuedefaultappendjoin isinstanceerrorsZOpErrormessager__name__ __class__Znode_defopZ error_code RuntimeErrorr)rr) rrr/Zbound_signaturerZarguments_contextargvaluer9Z display_nameZnew_er object_namerrr \sl=         <z8inject_argument_info_in_traceback..error_handler)rr%r&r')rrAr rr@r!inject_argument_info_in_tracebackMsDrBcCs,t|tjrd|jd|jjdSt|S)Nztf.Tensor(shape=z, dtype=))r7rZTensorshapeZdtyper2repr)r?rrrr3s r3r)__doc__r.osr!r rZtensorflow.compat.v2compatZv2rpathabspathr6__file__rrrr(rBr3rrrrs   V