o i e7@sjdZddlZddlZddlZddlmmZe edr ej Z ne dgdZ e edr1ej Z ne dgdZ ddZ e ed rJejZd d ZnejZd d ZddZddZddZdd ZddZddZddZddZd>ddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Z d/d0Z!d1d2Z"d3d4Z#d5d6Z$d7d8Z%d9d:Z&d?ds zgetargspec..N)r functoolspartial_get_argspec_for_partialtf __internal__ decoratorunwrapnextr TypeErrortype__init____new____call__)obj decoratorsrspecrrrrrs8        rc st|j}|jp i}t|j\}}}|dtgt}|r.||t| d<|D]\}}|vrD|} ||| <q2|sJtdq2t fddt |Dddurdt ||dSfddt |D} | r{td| dt ||t |dS) zImplements `getargspec` for `functools.partial` objects. Args: obj: The `functools.partial` object Returns: An `inspect.ArgSpec` Raises: ValueError: When callable's signature can not be expressed with ArgSpec. NzSFunction does not have **kwargs parameter, but contains an unknown partial keyword.c3s |] \}}|ur|VqdSrr)r!idxx) no_defaultrrr#sz+_get_argspec_for_partial..cs(g|]\}}|ur|kr|qSrr)r!ijr first_defaultr6rr s z,_get_argspec_for_partial..zSome arguments z| do not have default value, but they are positioned after those with default values. This can not be expressed with ArgSpec.) lenrrrfuncobjectitemsindex ValueErrorr+ enumeratertuple) r1 n_prune_argspartial_keywordsrrr all_defaultskwdefaultr4invalid_default_valuesrr9rr&s< *     r&cCs<tjj|\}}|D]}|jdurt|jSq t|S)aTFDecorator-aware replacement for `inspect.getfullargspec`. This wrapper emulates `inspect.getfullargspec` in[^)]* Python2. Args: obj: A callable, possibly decorated. Returns: The `FullArgSpec` that describes the signature of the outermost decorator that changes the callable's signature. If the callable is not decorated, `inspect.getfullargspec()` will be called directly on the callable. N)r'r(r)r*r rr)r1r2rr"rrrr s  c s|d}|dd}t|}|t|ddpt|dd}t|r)|r)|f|}fdd|jD}tt|||jsBdnt |j}|rbt|j| d|jD] \}} |vra| |<qU|j dury|j D] \} } | vrx| | <qlS)anTFDecorator-aware replacement for inspect.getcallargs. Args: *func_and_positional: A callable, possibly decorated, followed by any positional arguments that would be passed to `func`. **named: The named argument dictionary that would be passed to `func`. Returns: A dictionary mapping `func`'s named arguments to the values they would receive if `func(*positional, **named)` were called. `getcallargs` will use the argspec from the outermost decorator that provides it. If no attached decorators modify argspec, the final unwrapped target's argspec will be used. rrNim_self__self__csg|]}|vr|qSrr)r!arg call_argsrrr;8szgetcallargs..) rcopygetattrismethodrupdatedictziprr<r r?) func_and_positionalnamedr= positionalrthisremaining_positionals default_countrLvaluekvrrMr getcallargs!s.      r^cOstj|i|Sr)r getframeinfo)rkwargsrrrr_Hsr_cC t|S)a3TFDecorator-aware replacement for inspect.getdoc. Args: obj: An object, possibly decorated. Returns: The docstring associated with the object. The outermost-decorated object is intended to have the most complete documentation, so the decorated parameter is not unwrapped. )rgetdocr1rrrrbLs rbcCs:tjj|d}t|drd|jvr|jdSt|S)z2TFDecorator-aware replacement for inspect.getfile.r f_globals__file__)r'r(r)r*hasattrrdrgetfile)r1unwrapped_objectrrrrg[s    rgcCs t||S)z5TFDecorator-aware replacement for inspect.getmembers.)r getmembers)r1 predicaterrrriks ricCra)z4TFDecorator-aware replacement for inspect.getmodule.)r getmodulercrrrrkp rkcCra)z1TFDecorator-aware replacement for inspect.getmro.)rgetmro)clsrrrrmurlrmcCttjj|dS)z4TFDecorator-aware replacement for inspect.getsource.r)r getsourcer'r(r)r*rcrrrrpzrpcCro)z8TFDecorator-aware replacement for inspect.getsourcefile.r)r getsourcefiler'r(r)r*rcrrrrrrqrrcCro)z9TFDecorator-aware replacement for inspect.getsourcelines.r)rgetsourcelinesr'r(r)r*rcrrrrsrqrscCro)z4TFDecorator-aware replacement for inspect.isbuiltin.r)r isbuiltinr'r(r)r*rcrrrrtrqrtcCro)z2TFDecorator-aware replacement for inspect.isclass.r)risclassr'r(r)r*rcrrrrurqrucCro)z5TFDecorator-aware replacement for inspect.isfunction.r)r isfunctionr'r(r)r*rcrrrrvrqrvcCroz3TFDecorator-aware replacement for inspect.ismodule.r)risframer'r(r)r*rcrrrrxrqrxcCro)z6TFDecorator-aware replacement for inspect.isgenerator.r)r isgeneratorr'r(r)r*rcrrrryrqrycCro)z>TFDecorator-aware replacement for inspect.isgeneratorfunction.r)risgeneratorfunctionr'r(r)r*rcrrrrzsrzcCro)z3TFDecorator-aware replacement for inspect.ismethod.r)rrQr'r(r)r*rcrrrrQrqrQcCrorw)rismoduler'r(r)r*rcrrrr{rqr{cCro)z4TFDecorator-aware replacement for inspect.isroutine.r)r isroutiner'r(r)r*rcrrrr|rqr|rcCst|ddS)z0TFDecorator-aware replacement for inspect.stack.rNr)contextrrrrsrr)r)(__doc__ collectionsr$inspectrtensorflow.compat.v2compatv2r'rfr namedtuplerrrrrrrr&r^r_rbrgrirkrmrprrrsrtrurvrxryrzrQr{r|rrrrrs^     9`'